[leedcode 52] N-Queens II

Follow up for N-Queens problem.

Now, instead outputting board configurations, return the total number of distinct solutions.

public class Solution {

    //本题类似于N-Queens，这个更简单一些，只需要求出解法的个数即可，因此没有了print步骤

    //注意：isvalid函数的参数，以及函数中i的范围

    //      A[]的意义，可以代表棋盘的行

    int res;

    int A[];

    public int totalNQueens(int n) {

        A=new int[n];

        nqueens(0,n);

        return res;

        

    }

    public void nqueens(int cur,int n){

        if(cur==n){

            res++;

        }else{

            for(int i=0;i<n;i++){

                A[cur]=i;

                if(isValid(cur)){

                   nqueens(cur+1,n); 

                }

            }

        }

    }

    public boolean isValid(int cur){

        for(int i=0;i<cur;i++){

            if(A[i]==A[cur]||Math.abs(A[i]-A[cur])==cur-i)

                return false;

        }

        return true;

    }

}