设计一种算法,打印 N 皇后在 N × N 棋盘上的各种摆法,其中每个皇后都不同行、不同列,也不在对角线上。这里的“对角线”指的是所有的对角线,不只是平分整个棋盘的那两条对角线。
题目要求所有皇后必须满足这样一种排序规则:任意两个皇后不能在同一行,同一列,同一对角线上。在剑指Offer上有这样一段话:如果题目要求按一定规则对数字排序,可以先求出这些数字的全排列,然后判断这些排列是否满足条件。这道题也是这样,可以先通过回溯法求出所有皇后的全排序,然后判断是否满足题中的条件。但是,如果N是一个很大的值,这种方法用时较长,事实上,我们在求全排列的过程中,就可以用一些条件对全排列的情况加以限制。
直接上代码:
class Solution {
List<List<String>> ans = new ArrayList<>();
public List<List<String>> solveNQueens(int n) {
char[][] chessboard = new char[n][n];
back(n,0,chessboard);
return ans;
}
//回溯,从上到下,当行号等于棋盘最后一行行号,即row==n的时候,代表找到一个满足规则的排列
public void back(int n,int row,char[][] chessboard){
if(row==n){
ans.add(generateList(chessboard));
return;
}
for(int col =0;col<n;col++){
if(check(chessboard,row,col)){
chessboard[row][col]='Q';
back(n,row+1,chessboard);
chessboard[row][col]='\u0000';
}
}
}
public List<String> generateList(char[][] chessboard){
List<String> list = new ArrayList<>();
for(char[] c_array : chessboard){
StringBuilder sb = new StringBuilder();
for(char c : c_array){
if(c=='\u0000'){
sb.append('.');
}else{
sb.append('Q');
}
}
list.add(sb.toString());
}
return list;
}
public boolean check(char[][] chessboard,int row,int col){
//检查同列
for(int i=0;i<row;i++){
if(chessboard[i][col]=='Q'){
return false;
}
}
//检查对角线(右上)
for(int i=row,j=col;i>=0&&j<chessboard.length;i--,j++){
if(chessboard[i][j]=='Q'){
return false;
}
}
//检查反对角线(左上)
for(int i=row,j=col;i>=0&&j>=0;i--,j--){
if(chessboard[i][j]=='Q'){
return false;
}
}
return true;
}
}