python 解决八皇后问题

import random

#判断是否冲突的函数
def conflict(state,nextX):
nextY=len(state)
for i in range(nextY):
if abs(state[i]-nextX) in (0,nextY-i):
#abs(state[i]-nextX)等于0表示在同一行，abs(state[i]-nextX)等于nextY-i表示在同一列
return True
return False

def queens(num,state=()):
#假设剩最后一个皇后没有位置时（对应当前皇后），遍历所有位置返回没有冲突的位置
for pos in range(num):
if not conflict(state, pos):
if len(state)==num-1:
yield(pos,)
#当元组只有一个元素时应该加，
else:
for result in queens(num, state+(pos,)):
yield(pos,)+result

def prettyprint(solution):      #输出皇后问题的矩阵 ，Q表示皇后的位置
def line(pos,length=len(solution)):
return 'X'*(pos)+'Q'+'X'*(length-1-pos)
for pos in solution:
print line(pos)

if __name__=='__main__':
prettyprint(random.choice(list(queens(8))))
#由于queens生成多种解决方案，random模块的choice方法会从一个有序类型中随机选一个