python 实现八皇后算法

问题描述

有一个棋盘和8个要放在棋盘上的皇后，唯一的要求是皇后之间不能构成威胁。

状态表示

为了表示一个可能的解决方案，可以使用元组，每个元组中的元素标识对应行上的皇后的位置，如：state[0]=3标识第一行上的皇后在第4列，当在某一递归的层面时，只能知道上一行皇后的位置，因此需要一个长度小于8的状态的元组。

寻找冲突

定义一个寻找冲突的函数conflict，将已知皇后的state和下一个皇后的位置作为参数，用于判断下一个皇后的位置是否与之前已放置的位置冲突：

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

参数nextX表示下个皇后的水平位置（即x坐标或列），nextY代表垂直位置，这个函数对前面的每个皇后的位置做一个检查，如果下一个皇后和前面的皇后有相同的水平位置或在一条对角线上，就会发生冲突，返回True，如果没有发生冲突则返回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  result + (pos,)

def prettyprint(solution):
    def line(pos,length=len(solution)):
        return '.  '*(pos) + 'X  '+'.  '*(length-pos-1)
    for pos in  solution:
        print(line(pos))

if __name__ == '__main__':
    #print(list(queens(8)))
    #print(len(list(queens(8))))
    for solution in queens(8):
        prettyprint(solution)
        print('======================')