题目描述:n 皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上,并且使皇后彼此之间不能相互攻击。
给定一个整数 n,返回所有不同的 n 皇后问题的解决方案。
每一种解法包含一个明确的 n 皇后问题的棋子放置方案,该方案中 ‘Q’ 和 ‘.’ 分别代表了皇后和空位。
解题思路:和矩阵中找字符串的思路是一样的,同样是回溯法加修改访问状态,这是这里的访问状态需要考虑行、列以及左右对角线,因为是n*n的棋盘,然后放的是n个皇后,所以每一行每一列都只能有一个皇后,因此我们用遍历行的方式,这样就不用记录行的访问状态,代码如下:
class Solution:
def solveNQueens(self, n: int) -> List[List[str]]:
if n == 0:return [[]]
res = []
ans = [['.']*n for _ in range(n)]
put_col = [False] * n
put_ldiag = [False] * (2*n + 1)
put_rdiag = [False] * (2*n + 1)
def traceback(row):
if row == n:
res.append([''.join(ans[i]) for i in range(n)])
return
for j in range(n):
if put_col[j] or put_ldiag[n-1-j+row] or put_rdiag[row+j]:
continue
ans[row][j] = 'Q'
put_col[j] = put_ldiag[n-1-j+row] = put_rdiag[row+j] = True
traceback(row+1)
ans[row][j] = '.'
put_col[j] = put_ldiag[n-1-j+row] = put_rdiag[row+j] = False
traceback(0)
return res