[Leetcode]N皇后-Python递归回溯

[Leetcode]N皇后

Leetcode-N皇后

题目描述

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

上图为 8 皇后问题的一种解法。

给定一个整数 n，返回所有不同的 n 皇后问题的解决方案。
每一种解法包含一个明确的 n 皇后问题的棋子放置方案，该方案中 ‘Q’ 和 ‘.’ 分别代表了皇后和空位。

示例：

输入：4
输出：[
 [".Q..",  // 解法 1
  "...Q",
  "Q...",
  "..Q."],

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

提示：
皇后彼此不能相互攻击，也就是说：任何两个皇后都不能处于同一条横行、纵行或斜线上。

解题思路

• 参考Leetcode-N皇后
• 在此基础上加了一个列表表示

实现代码

class Solution:
    def solveNQueens(self, n: int) -> List[List[str]]:
        res = []
        ys = [0 for i in range(n)]
        def isvalid(x, y, ys):  # 第x行第y列是否有效，ys中存放每一行放置皇后的位置
            for i in range(x):
                if ys[i] == y or abs(i - x) == abs(y - ys[i]):
                    return False
            return True

        def backtrace(x, ans):
            if x == n:
                res.append(ans[:])
                return
            for i in range(n):
                if isvalid(x, i, ys):
                    ys[x] = i
                    ans[x] = "".join("." if j != i else "Q" for j in range(n))
                    backtrace(x + 1, ans)

        ans = ["." * n] * n
        backtrace(0, ans)
        return res