leetcode 51. N 皇后-java实现

题目所属分类

dfs板子题

原题链接

按照国际象棋的规则，皇后可以攻击与之处在同一行或同一列或同一斜线上的棋子。

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

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

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

题解

第y行 第x个 在y与x上写反了可以说
也可以参考这篇 Acwing排列数字、n皇后问题 java实现
这里面的dg和udg一般都是2N 要求的是N在10以内 为了方便 就统一定在20

class Solution {
     static int N = 20;
    static char[][] g = new char[N][N];
    static boolean[] col = new boolean[N];
    static boolean[] dg = new boolean[N];//正斜
    static boolean[] udg = new boolean[N];//反斜
    static List<List<String>> ans = new ArrayList<>();
  
    public List<List<String>> solveNQueens(int n) {
        ans.clear();
        for(int i = 0 ; i < n ; i++){
            for(int j = 0 ; j < n ; j++){
                g[i][j] = '.';            
            }
        }
        dfs(0,n);
        return ans ;
    }
    public void dfs(int y , int n ){
        if(y == n ){
             List<String> path = new ArrayList<>();
            for(int i = 0 ; i < n ; i++){
                String t = "";
                for(int j = 0 ; j < n ; j ++){
                    t += g[i][j];
                }
                path.add(t);
            } 
            ans.add(path);
            return  ;
        }
        for(int x = 0 ; x < n ; x++){
            if(!col[x] && !dg[y-x+n] & !udg[y+x]){
                g[y][x] = 'Q';
                col[x] =  dg[y - x + n] = udg[x + y] = true;               
                dfs(y + 1,n);
                col[x] =  dg[y - x + n] = udg[x + y] = false;   
                g[y][x] = '.';
            }
        }
    }
}