N-Queens II leetcode java

题目：

Follow up for N-Queens problem.

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

 

题解：

这道题跟NQueens的解法完全一样（具体解法参照N QueensN Queens leetcode java），只不过要求的返回值不同了。。所以要记录的result稍微改一下就好了。。。

因为涉及到递归，result传进去引用类型（List，数组之类的）才能在层层递归中得以保存，所以这里使用一个长度为1的数组帮助计数。

当然，也可以使用一个全局变量来帮助计数。

代码如下：

 

 1     public int totalNQueens(int n) {  
 2         int[] res = {0};
 3         if(n<=0)
 4             return res[0];
 5             
 6         int [] columnVal = new int[n];
 7         
 8         DFS_helper(n,res,0,columnVal);
 9         return res[0];
10     }
11     
12     public void DFS_helper(int nQueens, int[] res, int row, int[] columnVal){
13         if(row == nQueens){
14             res[0] += 1;
15         }else{
16             for(int i = 0; i < nQueens; i++){
17                 columnVal[row] = i;//(row,columnVal[row)==>(row,i)
18                 
19                 if(isValid(row,columnVal))
20                     DFS_helper(nQueens, res, row+1, columnVal);
21             }
22         }
23     }
24     
25     public boolean isValid(int row, int [] columnVal){
26         for(int i = 0; i < row; i++){
27             if(columnVal[row] == columnVal[i]
28                ||Math.abs(columnVal[row]-columnVal[i]) == row-i)
29                return false;
30         }
31         return true;

 使用全局变量来记录结果的代码是：

 1     int res;
 2     public int totalNQueens(int n) { 
 3         res = 0;
 4         if(n<=0)
 5             return res;
 6             
 7         int [] columnVal = new int[n];
 8         
 9         DFS_helper(n,0,columnVal);
10         return res;
11     }
12     
13     public void DFS_helper(int nQueens, int row, int[] columnVal){
14         if(row == nQueens){
15             res += 1;
16         }else{
17             for(int i = 0; i < nQueens; i++){
18                 columnVal[row] = i;//(row,columnVal[row)==>(row,i)
19                 
20                 if(isValid(row,columnVal))
21                     DFS_helper(nQueens, row+1, columnVal);
22             }
23         }
24     }
25     
26     public boolean isValid(int row, int [] columnVal){
27         for(int i = 0; i < row; i++){
28             if(columnVal[row] == columnVal[i]
29                ||Math.abs(columnVal[row]-columnVal[i]) == row-i)
30                return false;
31         }
32         return true;
33     }