LeetCode —— 面试题 08.12. 八皇后（Python）

设计一种算法，打印 N 皇后在 N × N 棋盘上的各种摆法，其中每个皇后都不同行、不同列，也不在对角线上。这里的“对角线”指的是所有的对角线，不只是平分整个棋盘的那两条对角线。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/eight-queens-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路：

class Solution:
    def __init__(self):
        self.res = []
    def solveNQueens(self, n: int) -> List[List[str]]:
        st = [['.' for i in range(n)] for j in range(n)]  #初始化n*n棋盘
        def dfs(x_d, y_d, cur):  # 深度优先遍历
        # x_d,y_d存储对角线结点信息
        # cur存储前j行的结点信息
            if len(cur) == n:  # 遍历停止条件
                m = []
                for i in st: 
                    m.append(''.join(i))
                self.res.append(m)
                return
            j = len(cur)  # 当前遍历到第j行
            for i in range(n):
                if i not in cur and i + j not in x_d and i-j not in y_d:
                # 判断是否已经出现在上面行中或者是否与上面的点是对角线
                    st[j][i] = 'Q'  # 改变当前结点位置
                    dfs(x_d+[i+j],y_d+[i-j],cur+[i])  # 查询下一层
                    st[j][i] = '.'  # 回溯
        dfs([], [], [])
        return res