LeetCode: N-Queens II 解题报告

最新推荐文章于 2022-06-06 10:35:33 发布

weixin_34205826

最新推荐文章于 2022-06-06 10:35:33 发布

阅读量134

点赞数

N-Queens II (LEVEL 4 难度级别，最高级5)

Follow up for N-Queens problem.

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

解答：
与第一题的解法是类似的，不同之处是，DFS的返回值是解的个数。我们每一次搜索本行8个位置时，
把8个位置的解都加在一起就行了。有的位置无解，它的返回值就是0咯。

另外注意BASE CASE:
当row == size时，也就是row越界了，这时意思是整个N-Queen都摆满了，我们这时应返回解为1.因为
你不需要任何动作，也就是唯一的解是保持原状。（虽然有点难以理解），同学可以退回到上一层来想
比如最后一行，你放置了一个皇后，它就是1个解，所以在那里dfs到下一层应该返回1.

 1 public class Solution {
 2     public int totalNQueens(int n) {
 3         if (n == 0) {
 4             return 0;
 5         }
 6         
 7         // Bug 1: forget to modify the parameters of the function.
 8         return dfs(n, 0, new ArrayList<Integer>());
 9     }
10     
11     public int dfs(int n, int row, ArrayList<Integer> path) {
12         if (row == n) {
13             // The base case: 当最后一行，皇后只有1种放法(就是不放)
14             return 1;
15         }
16         
17         int num = 0;
18         
19         // The queen can select any of the slot.
20         for (int i = 0; i < n; i++) {
21             if (!isValid(path, i)) {
22                 continue;
23             }
24             path.add(i);
25             
26             // All the solutions is all the possiablities are add up.
27             num += dfs(n, row + 1, path);
28             path.remove(path.size() - 1);
29         }
30         
31         return num;
32     }
33     
34     public boolean isValid(ArrayList<Integer> path, int col) {
35         int size = path.size();
36         for (int i = 0; i < size; i++) {
37             // The same column with any of the current queen.
38             if (col == path.get(i)) {
39                 return false;
40             }
41             
42             // diagonally lines.
43             // Bug 2: forget to add a ')'
44             if (size - i == Math.abs(col - path.get(i))) {
45                 return false;
46             }
47         }
48         
49         return true;
50     }
51 }