https://leetcode.com/problems/n-queens/description/
n 皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上,并且使皇后彼此之间不能相互攻击。皇后的走法:可以到达横向、纵向和斜线(45度,135度)的任意距离。
给定一个整数 n,返回所有不同的 n 皇后问题的解决方案。
每一种解法包含一个明确的 n 皇后问题的棋子放置方案,该方案中 ‘Q’ 和 ‘.’ 分别代表了皇后和空位。
思路:回溯。从第一行开始填,每次对0~n-1列遍历,检查每个位置是否合法,若合法则填入,并递归进入第二行。
用到了python的多变量for循环,用zip函数。注意循环的终点以范围最短的为准:
>>> for i,j,k in zip(range(9), range(1,9), range(2,11)): #每次对三个变量分别循环
... print(i,j,k)
...
0 1 2
1 2 3
2 3 4
3 4 5
4 5 6
5 6 7
6 7 8
7 8 9 #j先到达终点,所有变量一起结束
此外由于回溯棋盘时,string的元素不能修改,因此用一个二维字符列表,而不是一维string列表来表示棋盘。而最终结果要求一个一维string列表,因此得到的结果还需要转换,将字符列表转为string,用join函数:
https://blog.csdn.net/u013810296/article/details/57082163
class Solution:
def solveNQueens(self, n):
"""
:type n: int
:rtype: List[List[str]]
"""
record = []
board = [['.'] * n for _ in range(n)] #注意二维列表的创建方式
self.solve(record, board, 0, n)
return record
def solve(self, record, board, row, n):
if row == n: #得到一个结果
res = [] #记录一个结果
for i in range(n):
res.append(''.join(board[i])) #将棋盘每行转为一个string填入结果
record.append(res)
else:
for col in range(n):
if self.isValid(board, row, col, n): #若合法则填入
board[row][col] = 'Q'
self.solve(record, board, row + 1, n) #递归进入下一行
board[row][col] = '.' #还原现场,回溯
def isValid(self, board, row, col, n): #判断row,col位置是否合法
for i in range(n):
if board[i][col] == 'Q':
return False
for i, j in zip(range(row - 1, -1, -1), range(col - 1, -1, -1)):
if board[i][j] == 'Q':
return False
for i, j in zip(range(row - 1, -1, -1), range(col + 1, n)):
if board[i][j] == 'Q':
return False
return True