Leetcode 52. N-Queens II

Leetcode 52. N-Queens II_第1张图片
方法1: 其实52题应该作为nqueens系列题目的第一题,51题作为第二题,这样符合难易顺序。两道题基本一模一样,这道题我们要求有多少种摆放方法。用51题的代码也是可以的,但是这道题其实没有没要place queue,我们只需要用三个数组来表明现在哪些行哪些列哪些对角线不能用就行。这边给出源代码的解释链接。

public class Solution {
    int count = 0;
    public int totalNQueens(int n) {
        boolean[] cols = new boolean[n];     // columns   |
        boolean[] d1 = new boolean[2 * n];   // diagonals \
        boolean[] d2 = new boolean[2 * n];   // diagonals /
        backtracking(0, cols, d1, d2, n);
        return count;
    }
    
    public void backtracking(int row, boolean[] cols, boolean[] d1, boolean []d2, int n) {
        if(row == n) count++;

        for(int col = 0; col < n; col++) {
            int id1 = col - row + n;
            int id2 = col + row;
            if(cols[col] || d1[id1] || d2[id2]) continue;
            
            cols[col] = true; d1[id1] = true; d2[id2] = true;
            backtracking(row + 1, cols, d1, d2, n);
            cols[col] = false; d1[id1] = false; d2[id2] = false;
        }
    }
}

总结:

  • 51和52题是非常经典的backtracking问题,需要好好掌握精髓。

你可能感兴趣的:(Leetcode,backtracking)