力扣面试题 08.12. 八皇后 Java实现

    设计一种算法，打印 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;
    }
}