n 皇后问题 研究的是如何将 n
个皇后放置在 n×n
的棋盘上,并且使皇后彼此之间不能相互攻击。
给你一个整数 n
,返回所有不同的 n 皇后问题 的解决方案。
每一种解法包含一个不同的 n 皇后问题 的棋子放置方案,该方案中 'Q'
和 '.'
分别代表了皇后和空位。
示例 1:
输入:n = 4 输出:[[".Q..","...Q","Q...","..Q."],["..Q.","Q...","...Q",".Q.."]] 解释:如上图所示,4 皇后问题存在两个不同的解法。
示例 2:
输入:n = 1 输出:[["Q"]]
提示:
1 <= n <= 9
以下程序实现了这一功能,请你填补空白处内容:
class Solution(object): def solveNQueens(self, n): if n == 0: return 0 res = [] board = [['.'] * n for t in range(n)] self.do_solveNQueens(res, board, n) return res def do_solveNQueens(self, res, board, num): if num == 0: res.append([''.join(t) for t in board]) return ls = len(board) pos = ls - num check = [True] * ls for i in range(pos): for j in range(ls): if board[i][j] == 'Q': check[j] = False step = pos - i if j + step < ls: check[j + step] = False if j - step >= 0: check[j - step] = False break for j in range(ls): if check[j]: board[pos][j] = 'Q' self.do_solveNQueens(res, board, num - 1) board[pos][j] = '.' if __name__ == '__main__': s = Solution() print (s.solveNQueens(4))