0 # -*- coding: utf-8 -*-
1 import random
2 #冲突检查，在定义state时，采用state来标志每个皇后的位置，其中索引用来表示横坐标，基对应的值表示纵坐标，例如： state[0]=3，表示该皇后位于第1行的第4列上
3 def conflict(state, nextX):
4 nextY = len(state)
5 for i in range(nextY):
6 #如果下一个皇后的位置与当前的皇后位置相邻（包括上下，左右）或在同一对角线上，则说明有冲突，需要重新摆放
7 if abs(state[i]-nextX) in (0, nextY-i):
8 return True
9 return False
10
11 #采用生成器的方式来产生每一个皇后的位置，并用递归来实现下一个皇后的位置。
12 def queens(num, state=()):
13 for pos in range(num):
14 if not conflict(state, pos):
15 #产生当前皇后的位置信息
16 if len(state) == num-1:
17 yield (pos, )
18 #否则，把当前皇后的位置信息，添加到状态列表里，并传递给下一皇后。
19 else:
20 for result in queens(num, state+(pos,)):
21 yield (pos, ) + result
22
23
24 #为了直观表现棋盘，用X表示每个皇后的位置
25 def prettyprint(solution):
26 def line(pos, length=len(solution)):
27 return '. ' * (pos) + 'X ' + '. '*(length-pos-1)
28 for pos in solution:
29 print line(pos)
30
31 if __name__ == "__main__":
32 prettyprint(random.choice(list(queens(8))))
