[LeetCode][Java] N-Queens II

Posted 2020-09-18 claireyuancy

题目：

Follow up for N-Queens problem.

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

题意：

接上题N-Queens problem，这里仅仅是要求返回全部的不同的解的数目。

算法分析：

思路与上题同样，代码比上题更加简洁了

AC代码：

<span style="font-size:12px;">public class Solution 
{
    private int res=0;
    public int totalNQueens(int n) 
    {  
        int[] loc = new int[n];  //记录皇后处于哪一列，列数组
        dfs(loc,0,n);  
        return res;  
    }  
    public void dfs(int[] loc, int cur, int n)
    {  
        if(cur==n)   
            res++;  
        else
        {  
            for(int i=0;i<n;i++)
            {  
                loc[cur] = i;  
                if(isValid(loc,cur))  
                    dfs(loc,cur+1,n);  //再放皇后m+1, 假设皇后m+1放完并返回了  
                                          //两种可能：  1：冲突，返回了  2.一直将所有的皇后所有放完并安全返回了  
                                        //将皇后m回溯。探索新的可能或者安全的位置  --->
            }  
        }  
    }  
    public boolean isValid(int[] loc, int cur)
    {  
        for(int i=0;i<cur;i++)//仅仅须要保证与那些已经就位的皇后不冲突就可以  
        {  
            if(loc[i]==loc[cur]||Math.abs(loc[i]-loc[cur])==(cur-i)) //验证对角线，依据对角线性质，长 ＝ 宽 
                                                                     //那么我们不难写出 Math.abs(loc[i] - loc[cur]) == (cur - i) 
                return false;  
        }  
        return true;  
    }  

}</span>