表示状态
使用元组来表示皇后的位置。如果state[0]==1,就表示皇后在第一行的第二列。
寻找冲突
冲突是什么?如果下一个皇后和前面的皇后在同样的水平位置,或者在同一条对角线上,就会发生冲突。观察八皇后的格局可以发现,所有的‘\’对角线上,皇后的坐标相等。所有的‘/’对角线上坐标相加的值相等。
由此可以定义一个判断下一个皇后是否冲突的函数
def conflict(state,nextX):
nextY = len(state)
#判断是否在同一水平线上
if nextX in state: return True
for i in range(nextY):
#判断是否在一条'\'线上
if (nextY-nextX == i-state[i]) : return True
#判断是否在一条'/'线上
if (nextY+nextX == i+state[i]) : 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
第3行判断是否冲突,第4行判断是否是最后一行,如果是,直接拼接上去。执行到第七行,有一个for代码块,它的含义是如果在pos这个位置不冲突,那就把皇后放在这里,然后依次寻找后面行的不冲突皇后。
关于yield
包含yield关键字的函数在Python里叫做生成器。生成器是逐渐产生结果的复杂递归算法的理想实现工具。yield类似于return都是返回值,但是yield返回一个值后不会停止,直到生成所有符合条件的值才停止。
输出
for i in queens(8):
print i