首页 > 代码库 > python学习八皇后问题

python学习八皇后问题

 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))))

 

python学习八皇后问题