Leetcode 51. N 皇后

题目链接
51. N 皇后

题目描述
n 皇后问题 研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。需要满足同一行、列和同一对角线上只能有一个皇后。
给你一个整数 n ，返回所有不同的 n 皇后问题 的解决方案。
每一种解法包含一个不同的 n 皇后问题 的棋子放置方案，该方案中 ‘Q’ 和 ‘.’ 分别代表了皇后和空位。

输入：n = 4
输出：[[".Q..","...Q","Q...","..Q."],["..Q.","Q...","...Q",".Q.."]]

实现
回溯，对于每一行，从第 0 列开始，如果没有同列且同对角线的 Q ，则假设 Q 在此列，如果有冲突或没有答案则回退到上一行，没有再看下一行，直到最后一行结束，得到一个结果。

class Solution:
    def solveNQueens(self, n: int) -> List[List[str]]:
        self.res = []
        # 对角线，2n-1个 \ 对角线 i-j相同，和2n-1个 / 对角线 i+j相同
        self.dia1 = [0 for _ in range(2*n-1)]
        self.dia2 = [0 for _ in range(2*n-1)]
        # 同列没有
        self.col = [0 for _ in range(n)]

        # path[i] = 第i行有Q的列
        self.dfs(n, [], 0)

        return self.res

    def dfs(self, n, path, index):
        # 到最后一行, 找到一组终止
        if index == n:
            board = [['.' for _ in range(n)] for _ in range(n)]
            for i in range(n):
                board[i][path[i]] = 'Q'
            self.res.append([''.join(x) for x in board])
        
        for j in range(n):
            # 尝试将index行的Q放在第i列，需要同列和对角线没有
            # i-j的坐标用i-j+n-1表示，i-j最小为-(n-1) 向后平移n-1位到0
            if not self.col[j] and not self.dia1[index+j] and not self.dia2[index-j+n-1]:

                self.col[j], self.dia1[index+j], self.dia2[index-j+n-1] = 1, 1, 1
                self.dfs(n, path+[j], index+1)
                self.col[j], self.dia1[index+j], self.dia2[index-j+n-1] = 0, 0, 0