Leetcode刷题记录 52. N皇后 II

n 皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。

给定一个整数 n，返回所有不同的 n 皇后问题的解决方案的个数。

每一种解法包含一个明确的 n 皇后问题的棋子放置方案，该方案中 'Q' 和 '.' 分别代表了皇后和空位。

示例:

输入: 4

输出: 2
解释: 4 皇后问题存在如下两个不同的解法。
[
 [".Q..",  // 解法 1
  "...Q",
  "Q...",
  "..Q."],

 ["..Q.",  // 解法 2
  "Q...",
  "...Q",
  ".Q.."]
]

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

class Solution(object):
    def solveNQueens(self, n: int) -> int:
        res = []
        if n == 0:
            return res
        # row为当前行，col为不可再使用的列，master为不可再使用的主对角线，slave为不可再使用的副对角线
        def dfs(row, col, master, slave, cur_res, n):
            if row == n:
                res.append(["." * cur + "Q" + "." * (n - cur - 1) for cur in cur_res])
                return

            for i in range(n):
                if (i not in col) and (i+row not in slave) and (i-row not in master):
                    dfs(row+1, col|{i}, master|{i-row}, slave|{i+row}, cur_res+[i], n)

        dfs(0, set(), set(), set(), [], n)
        return len(res)   

        

if __name__=="__main__":
    print(Solution().solveNQueens(4))