八皇后问题的一种python解法

八皇后问题，是一个古老而著名的问题，是回溯算法的典型例题。该问题是十九世纪著名的数学家高斯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))))