n 皇后问题 研究的是如何将 n 个皇后放置在 n×n 的棋盘上,并且使皇后彼此之间不能相互攻击。
给你一个整数 n ,返回所有不同的 n 皇后问题 的解决方案。
每一种解法包含一个不同的 n 皇后问题 的棋子放置方案,该方案中 'Q' 和 '.' 分别代表了皇后和空位。
示例 1:
输入:n = 4
输出:[[".Q..","...Q","Q...","..Q."],["..Q.","Q...","...Q",".Q.."]]
解释:如上图所示,4 皇后问题存在两个不同的解法。
示例 2:
输入:n = 1 输出:[["Q"]]
解题思路
以递归方式,依次在每一行进行n个位置皇后放置,当放置第n+1个皇后时,说明前n个皇后以安全放置
代码
class Solution:
def solveNQueens(self, n: int) -> List[List[str]]:
self.pos=[]
self.pos_queen(0,n,[])
res=[]
for data in self.pos:
cur_res=[]
# cur_str='.' * n
for x, y in data:
cur_str = ['.'] * n
cur_str[y]="Q"
cur_res.append("".join(cur_str))
res.append(cur_res)
return res
def pos_queen(self,index,n,pos=[]):
if index>=n:
# print(pos)
self.pos.append(pos)
#尝试放n+1个皇后,说明前n个安全
return True
for i in range(n):
if self.safe_pos_check(index,i,pos):
cur_pos=copy.deepcopy(pos)
cur_pos.append([index,i])
self.pos_queen(index+1,n,cur_pos)
def safe_pos_check(self,x,y,pos):
if not pos:
return True
for xi,yi in pos:
if xi==x or yi==y or abs(xi-x)==abs(yi-y):
return False
return True