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