题目:
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 }