题目
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皇后约束条件:
- 不能同行
- 不能同列
- 不能同斜线
仍然可以把二维棋盘抽象为一棵树:(以3皇后为例举例说明)
可以看到,二维矩阵的高就是树的高度,矩阵的宽就是树中每个节点宽度
回溯三部曲:
- 递归函数参数:二维数组result存放结果,参数n记录棋盘的大小(矩阵的长宽),用row记录遍历到第几行了
- 递归终止条件:就像上面我画的图,递归到叶子节点的时候就收集到了一个结果了
if(row == n){
result.add(chessboard);
return;
}
- 单层搜索逻辑:递归深度就是用row控制棋盘的行,每一层for循环的col控制棋盘的列,一行一列,从而确定皇后的位置
- 验证棋盘是否合法:即一下三个条件
1、不能同行(这个其实不用检查,因为在每层递归中,只会选择for循环(同一行)中的一个元素,所以不需要去重操作)
2、不能同列
3、不能同斜线
java代码如下:
class Solution {
List<List<String>> res = new ArrayList<>();
public List<List<String>> solveNQueens(int n){
char[][] chessboard = new char[n][n];//初始化存储棋盘的二维字符数组
for(char[] c : chessboard){
Arrays.fill(c,'.');//初始全部用.来填满棋盘
}
backTrack(n, 0, chessboard);
return res;
}
public void backTrack(int n, int row, char[][] chessboard){
if(row == n){//递归到叶子节点时,收集结果
res.add(Array2List(chessboard));//将符合格式的结果加入到res中
return;
}
for(int col = 0; col < n; col++){//row控制行,col控制列
if(isValid(row, col, n, chessboard)){
chessboard[row][col] = 'Q';
backTrack(n, row+1, chessboard);
chessboard[row][col] = '.';//回溯回之前的状态
}
}
}
public List Array2List(char[][] chessboard){
List<String> list = new ArrayList<>();
for(char[] c : chessboard){
list.add(String.copyValueOf(c));//copyValueOf()是Java中的String方法,用于创建具有给定字符数组的字符串,即,它接受一个字符数组并返回一个字符串。 换句话说,可以说copyValueOf()方法用于将字符数组的值复制到字符串,比如把二维数组中的字符'.','.','.','Q'四个字符拼接成了一个字符串“...Q”并且存在list中,即["...Q"]
}
return list;//返回的列表相当于一个满足格式的结果
}
public boolean isValid(int row, int col, int n, char[][] chessboard){
//检查是否同一列,row控制行,col控制列,自己画图理解即可
for(int i = 0; i < row; i++){
if(chessboard[i][col] == 'Q'){
return false;
}
}
//检查右下到左上的斜对角线
for (int i = row-1, j = col-1; i >= 0 && j >= 0; i--, j--) {
if (chessboard[i][j] == 'Q') {
return false;
}
}
//检查左下到右上的斜对角线
for (int i = row-1, j = col + 1; i >= 0 && j <= n-1; i--, j++) {
if (chessboard[i][j] == 'Q') {
return false;
}
}
return true;
}
}