八皇后问题,是一个古老而著名的问题,是回溯算法的典型案例。在8X8格的国际象棋上摆放八个皇后,使其不能互相攻击,即任意两个皇后都不能处于同一行、同一列或同一斜线上,有92种摆法。
def conflict(state, nextX):
nextY = len(state)
for i in range(nextY):
if abs(state[i] - nextX) in (0, nextY - 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
#输出图像的函数
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))))
可以得到如下图形:
. . X . . . . .
. . . . X . . .
. X . . . . . .
. . . . . . . X
. . . . . X . .
. . . X . . . .
. . . . . . X .
X . . . . . . .