python N皇后问题

题目：

需要将N个皇后放在N*N的棋盘上，条件是任何一个皇后都不能威胁其他皇后，即任何两个皇后都不能吃掉对方。怎样才能做到这一点呢？已将这些皇后放在什么地方呢？

思路：

1.以第一个和第二个皇后为例，第一个皇后默认在第一行，第二个皇后在第二行，二者列不相等，且不在斜线上。
2.record[] 用来存储第i个皇后的列
3.设第i个皇后的位置为 （i，record[i]）下一个摆放位置为（a，b），当满足|a-i|==|b-record[i]|,说明在一个斜线上，则（a，b） 这个点不能作为下一个插入的点。只有当（a，b）满足于之前i个皇后节点的列均不相等，且均不在同一斜线上时可以插入
4.把递归过程直接设计成逐行放置皇后的方式，可以避开1的那些不能放置的位置。

class Solution:
    def judge(self, record, i, j):## i 代表第几个皇后， j表示第i个皇后的列
        for k in range(0, i):
            if j == record[k] or  abs(record[k] - j) == abs(i-k):
                return False
        return True


    def queen(self, record, i, n):#i代表第I个皇后
        if i == n:
            return 1
        res = 0

        for j in range(0, n):
            if self.judge(record, i, j):
                record[i] = j

                res += self.queen(record, i+1, n)

        return res

a = Solution()
record = [0]*8
b = a.queen(record, 0, 8)
print(b)