题目:
按照国际象棋的规则,皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。
n 皇后问题 研究的是如何将 n
个皇后放置在 n×n
的棋盘上,并且使皇后彼此之间不能相互攻击。
给你一个整数 n
,返回所有不同的 n 皇后问题 的解决方案。
每一种解法包含一个不同的 n 皇后问题 的棋子放置方案,该方案中 'Q'
和 '.'
分别代表了皇后和空位。
思路:棋盘的长就是for循环的长度,棋盘的宽就是递归的深度,这样就可以套进回溯法的模板里了。
第一步:确定参数与返回值。参数为棋盘行/列数n,树的深度row,棋盘局势chessboard,无返回值
第二步:确定终止条件。当row=n时,说明到达叶子节点,将结果存入结果集中
第三步:确定单层递归逻辑。for循环遍历棋盘每一列,如果棋盘局势合法,则将(row,col)坐标的元素设为Q,再递归回溯
代码:
List<List<String>> result=new ArrayList<>();
public List<List<String>> solveNQueens(int n) {
char[][] chessboard=new char[n][n];
for(char[] c:chessboard){
Arrays.fill(c,'.');
}
backTracking(n,0,chessboard);
return result;
}
//递归,n为棋盘行/列大小,row为树的深度同时也是棋盘的行,chessboard为棋盘
public void backTracking(int n,int row,char[][] chessboard){
if(row==n){
result.add(Array2List(chessboard));
return;
}
for(int col=0;col<n;col++){
if(isValid(row,col,n,chessboard)){
chessboard[row][col]='Q';//处理
backTracking(n,row+1,chessboard);//递归
chessboard[row][col]='.';//回溯
}
}
}
//检查棋盘的合法性,row为棋盘的行,col为棋盘的列,n为棋盘的行/列大小,chessboard为棋盘的情况
public boolean isValid(int row,int col,int n,char[][] chessboard){
//树的一层为棋盘的一行,因此不用检查行
//条件一:不为同一列,检查列
for(int i=0;i<row;i++){
if(chessboard[i][col]=='Q')
return false;
}
//条件二:不为同一斜线,检查45度对角线
for(int i=row-1,j=col-1;i>=0&&j>=0;i--,j--){
if(chessboard[i][j]=='Q')
return false;
}
//条件二:不为同一斜线,检查135度对角线
for(int i=row-1,j=col+1;i>=0&&j<n;i--,j++){
if(chessboard[i][j]=='Q')
return false;
}
return true;
}
//将每一行的内容转换为string
public List Array2List(char[][] chessboard){
List<String> list=new ArrayList<>();
for(char[] c:chessboard){
list.add(String.copyValueOf(c));
}
return list;
}