关键词:python,八皇后,生成器,迭代
def conflict(state,nextX):
nextY=len(state)
for i in range(nextY):
if abs(state[i]-nextY) in (0,nextY-1):
return True
return False
def queens(num=8,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):
def line(pos,length=len(solution)):
return '. '*(pos)+'X '+'. '*(length-pos-1)
for pos in solution:
print(line(pos))
import random
prettyprint(random.choice(list(queens(8))))
关键点:
1.怎么理解冲突条件:
abs(state[i]-nextY) in (0,nextY-1)
水平距离为0,表示在同一列;水平距离等于垂直距离,表示在对角线。
2.生成器的用法:
yield