[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