八皇后问题,是一个古老而著名的问题,是回溯算法的典型例题。该问题是十九世纪著名的数学家高斯1850年提出:在8X8格的国际象棋上摆放八个皇后,使其不能互相攻击,即任意两个皇后都不能处于同一行、同一列或同一斜线上,问有多少种摆法。 高斯认为有76种方案。1854年在柏林的象棋杂志上不同的作者发表了40种不同的解,后来有人用图论的方法解出92种结果。计算机发明后,有多种方法可以解决此问题。
回溯算法的基本思想是:从一条路往前走,能进则进,不能进则退回来,换一条路再试。八皇后问题就是回溯算法的典型,第一步按照顺序放一个皇后,然后第二步符合要求放第2个皇后,如果没有位置符合要求,那么就要改变第一个皇后的位置,重新放第2个皇后的位置,直到找到符合条件的位置就可以了。
本解法来自<<Python基础教程>>第三版。
注:原代码中并无print()语句
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
File Name: queens.py
Time : 2018/9/24 22:57
IDE :PyCharm
Author :Administrator
"""
import random
def conflict(state, nextX):
nextY = len(state)
print("将要检测第{}行,{}列的位置是否与既有皇后位置为{}冲突".format(nextY, nextX, state))
for i in range(nextY):
if abs(state[i] - nextX) in (0, nextY - i):
print("\033[7;31;40m检测结果{}行{}列的位置与既有皇后中第{}行{}列位置的皇后冲突\033[0m".format(nextY, nextX, i, state[i]))
return True
print("\033[7;32;40m检测结果:{}行{}列的位置与既有皇后位置{}不冲突\033[0m".format(nextY, nextX, state))
return False
def queens(num=8, state=()):
for pos in range(num):
if not conflict(state, pos):
print("\033[7;36;40m既有皇后位置为{},第{}行皇后的合法位置位于{}列\033[0m".format(state, len(state), pos))
if len(state) == num - 1:
print("第0行至第{}行的皇后位置为{},最后一行皇后位置的横坐标为{}".format(num - 2, state, pos))
yield (pos,)
else:
print("当前皇后位于第{}行,将进入下一轮递归".format(len(state)))
print("下一轮递归中的已知既有皇后的位置为{}".format(state + (pos,)))
for result in queens(num, state + (pos,)):
print("\033[7;44;40m既有皇后位置为{}\033[0m".format(state + (pos,)))
print("\033[7;44;40m当前行下面皇后位置为{}\033[0m".format(result))
yield (pos,) + result
def prettyprint(solution):
def line(pos, length=len(solution)):
return "."*pos + "X" + "."*(len(solution)-pos-1)
for pos in solution:
print(line(pos))
prettyprint(random.choice(list(queens(8))))