力扣51. N 皇后（JAVA）

51. N 皇后

力扣题解
难度困难1140

n 皇后问题 研究的是如何将 n 个皇后放置在 n×n 的棋盘上，并且使皇后彼此之间不能相互攻击。

给你一个整数 n ，返回所有不同的 n 皇后问题 的解决方案。

每一种解法包含一个不同的 n 皇后问题 的棋子放置方案，该方案中 'Q' 和 '.' 分别代表了皇后和空位。

示例 1：

输入：n = 4
输出：[[".Q..","...Q","Q...","..Q."],["..Q.","Q...","...Q",".Q.."]]
解释：如上图所示，4 皇后问题存在两个不同的解法。

示例 2：

输入：n = 1
输出：[["Q"]]

解析：套用回溯框架

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-buCK87zi-1641433611594)(C:\Users\WXY\AppData\Roaming\Typora\typora-user-images\image-20220106094511709.png)]

代码：

class Solution {
    List<List<String>> res=new ArrayList<>();
    public List<List<String>> solveNQueens(int n) {
        char[][] board=new char[n][n];
        for(char[] c:board){
            Arrays.fill(c,'.');
        }
        backtrack(n,0,board);
        return res;
    }
    void backtrack(int n,int row,char[][] board){
        if(row==n){
            res.add(ArrayList2(board));
        }
        for(int col=0;col<n;col++){
            if(!isVaild(board,row,col,n)){
                continue;
            }
            board[row][col]='Q';
            backtrack(n,row+1,board);
            board[row][col]='.';
        }
    }
    public List ArrayList2(char[][] board){
        List<String> list=new ArrayList<>();
        for(char[] c:board){
            list.add(String.copyValueOf(c));
        }
        return list;
    }
    boolean isVaild(char[][] board,int row,int col,int n){
        //检查列是否有冲突
        for(int i=0;i<n;i++){
            if(board[i][col]=='Q') return false;
        }
        //检查左上方是否有皇后互相冲突
        for(int i=row-1,j=col-1;i>=0 && j>=0;i--,j--){
            if(board[i][j]=='Q') return false;
        }
        //检查右上方是否有冲突
        for(int i=row-1,j=col+1;i>=0 && j<n;i--,j++){
            if(board[i][j]=='Q') return false;
        }
        return true;
    }
}