题目
n 皇后问题研究的是如何将 n 个皇后放置在 n×n 的棋盘上,并且使皇后彼此之间不能相互攻击。
皇后不能在同一行,同一列,同对角线
思路一 回溯思想
注意:
- 一行只可能有一个皇后且一列也只可能有一个皇后。这意味着没有必要再棋盘上考虑所有的方格。只需要按列循环即可。
思路步骤:
-
写回溯函数 helper(row = 0).
-
从第一个 row = 0 开始.
-
循环列并且试图在每个 column 中放置皇后.
-
判断方格 (row, column) 能不能放Q
-
在 (row, column) 方格上放置Q。
-
If 所有的行被考虑过,row == N
意味着我们找到了一个解 -
Else
继续考虑接下来的皇后放置 backtrack(row + 1).
回溯:将在 (row, column) 方格的Q移除.
class Solution {
public List<List<String>> solveNQueens(int n) {
List<List<String>> res = new ArrayList<>();
char[][] board = new char[n][n];
init(board);
helper(res, board, 0);
return res;
}
//初始化这个board,都为 .
private void init(char[][] board) {
for (int i = 0; i < board.length; i++) {
Arrays.fill(board[i], '.');
}
}
private void helper(List<List<String>> res, char[][] board, int rowIndex) {
if (rowIndex == board.length) { //递归终止条件,行数列数都是0开始,达到length,就是越界了
res.add(generate(board));
return;
}
for (int i = 0; i < board.length; i++) { //当前行的每一列都去判断是否能放Q
if (isAllow(board, rowIndex, i)) {
board[rowIndex][i] = 'Q';
helper(res, board, rowIndex + 1);
board[rowIndex][i] = '.'; //回溯,删除前面的那个动作,就是把加入的Q变成.,然后在上一次继续去尝试寻找答案
}
}
}
//判断当前位置是否能放置Q
private boolean isAllow(char[][] board, int rowIndex, int colIndex) {
for (int i = 0; i < rowIndex; i++) {
if (board[i][colIndex] == 'Q') {
return false;
}
}
for (int i = rowIndex - 1, j = colIndex - 1; i >= 0 && j >= 0; i--, j--) {
if (board[i][j] == 'Q') {
return false;
}
}
for (int i = rowIndex - 1, j = colIndex + 1; i >= 0 && j < board.length; i--, j++) {
if (board[i][j] == 'Q') {
return false;
}
}
return true;
}
//将board这个字符串数组变成链表
private List<String> generate(char[][] board) {
List<String> list = new ArrayList<>();
for (char[] row : board) {
StringBuilder sb = new StringBuilder();
for (char c : row) {
sb.append(c);
}
list.add(sb.toString());
}
return list;
}
}
时间复杂度:O(N!). 放置第 1 个皇后有 N 种可能的方法,放置两个皇后的方法不超过 N (N - 2) ,放置 3 个皇后的方法不超过 N(N - 2)(N - 4) ,以此类推。总体上,时间复杂度为O(N!) .
空间复杂度:O(N^2) ,建立一个char[][]
二维数组