leetcode 52. N-Queens II

题目
和上一题的思路一样，只不过变成统计个数。

class Solution {
    public int totalNQueens(int n) {
        int[] res = new int[1];
        String[][] mat = new String[n][n];
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                mat[i][j] = ".";
            }
        }
        dfs(mat,0,n,res);
        return res[0];
    }
    public void dfs(String[][] mat,int pos,int n,int[] res){
        if(pos==n-1){
            for(int i=0;i<n;i++){
                if(isVaild(mat,pos,i)) res[0]++;
            }
            
        }else{
            for(int i=0;i<n;i++){
                if(isVaild(mat,pos,i)){
                    mat[pos][i] = "Q";
                    dfs(mat,pos+1,n,res);
                    mat[pos][i] = ".";
                }
            }
        }
    }
    public boolean isVaild(String[][] mat,int i,int j){
        int row = i;
        while(row>=0){
            if(mat[row][j].equals("Q")) return false;
            row--;
        }
        int left=j,up=i;
        while(left>=0 && up>=0){
            if(mat[up][left].equals("Q")) return false;
            left--;
            up--;
        }
        int right=j,up1=i;
        while(right<mat.length && up1>=0){
            if(mat[up1][right].equals("Q")) return false;
            up1--;
            right++;
        }
        return true;
    }
}

直接用res是传递不了值的，所以采用了数组。