51. N 皇后
按照国际象棋的规则,皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。
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 {
// 存储所有合法的放置方案
List<List<String>> res = new ArrayList<>();
/**
* 输入棋盘边长 n,返回所有合法的放置
*/
public List<List<String>> solveNQueens(int n) {
// 初始化空棋盘,'.' 表示空,'Q' 表示皇后。注意board的每一行都是一个字符串而不是一个List。
List<String> board = new ArrayList<>();
StringBuilder sb = new StringBuilder();
for (int i = 0; i < n; i++) {
sb.append('.'); // 每行初始化为 '....'
}
for (int i = 0; i < n; i++) {
board.add(sb.toString()); // 每行都是 '....'
}
// 开始回溯
backtrack(board, 0);
return res;
}
/**
* 路径:board 中小于 row 的那些行都已经成功放置了皇后
* 选择列表:第 row 行的所有列都是放置皇后的选择
* 结束条件:row 超过 board 的最后一行
*/
private void backtrack(List<String> board, int row) {
// 结束条件:如果当前行是最后一行,说明找到了一个合法的放置方案
if (row == board.size()) {
res.add(new ArrayList<>(board)); // 将当前棋盘配置添加到结果列表中
return;
}
int n = board.get(row).length();
// 遍历当前行的列元素
for (int col = 0; col < n; col++) {
// 排除不合法选择
if (!isValid(board, row, col)) {
continue;
}
char[] arr = board.get(row).toCharArray();
// 下面两行代码是做选择:在 (row, col) 位置放置皇后
arr[col] = 'Q';
board.set(row, String.valueOf(arr));
// 进入下一行决策
backtrack(board, row + 1);
// 撤销选择:将 (row, col) 位置的皇后移除
arr[col] = '.';
board.set(row, String.valueOf(arr));
}
}
// 是否可以在 board[row][col] 放置皇后?因为isValid中的row参数是递增的,即我们是从上到下放置的,所以只需要检查上面行的元素是否和当前元素冲突即可。
private boolean isValid(List<String> board, int row, int col) {
int n = board.size();
// 检查之前行的同一列是否有皇后
for (int i = 0; i <= row; i++) {
if (board.get(i).charAt(col) == 'Q') {
return false; // 如果同列上有皇后,返回 false
}
}
// 检查右上斜线上是否有皇后互相冲突
for (int i = row - 1, j = col + 1; i >= 0 && j < n; i--, j++) {
if (board.get(i).charAt(j) == 'Q') {
return false; // 如果右上方有皇后,返回 false
}
}
// 检查左上斜线上是否有皇后互相冲突
for (int i = row - 1, j = col - 1; i >= 0 && j >= 0; i--, j--) {
if (board.get(i).charAt(j) == 'Q') {
return false; // 如果左上方有皇后,返回 false
}
}
return true; // 如果上述检查都通过,返回 true
}
}
- 时间复杂度:
O
(
n
2
⋅
n
!
)
O(n^2⋅n!)
O(n2⋅n!)。这和排列问题有点像,搜索树中至多有 O(n!) 个叶子,每个叶子生成答案每次需要
O
(
n
2
)
O(n^2)
O(n2) 的时间,即
res.add(new ArrayList<>(board))
需要 O ( n 2 ) O(n^2) O(n2) 的时间。所以时间复杂度为 O ( n 2 ⋅ n ! ) O(n^2⋅n!) O(n2⋅n!)。实际上搜索树中远没有这么多叶子,n=9 时只有 352 种放置方案,远远小于 9!=362880。更加准确的方案数可以参考 OEIS A000170,为 O ( n ! 2.5 4 n ) O(\frac{n!}{2.54^n}) O(2.54nn!)。 - 空间复杂度:O(n)。返回值的空间不计入。