LeetCode 52. N-Queens II

问题描述

• 地址

问题分析

• 该题其实比 LeetCode 51. N-Queens 要简单，因为 51 是返回所有可能性具体情况，而该题只要统计可能性次数即可。
• 大致流程和 LeetCode 51. N-Queens 一样，只不过如何返回值，可以有很多方法：
  
  • 声明一个全局变量count来统计
  • 传入一个count数组，用作指针的作用
  • 将dfs函数返回值改成int，返回

经验教训

• 在DFS中，如何统计可能性的个数

代码实现

• 声明一个全局变量count来统计

    //声明个全局变量，来统计个数
    int count = 0;
    public int totalNQueens(int n) {
        if (n <= 0) {
            return 0;
        }
        boolean[] d1 = new boolean[2 * n - 1];
        boolean[] d2 = new boolean[2 * n - 1];
        boolean[] colVisited = new boolean[n];
        findQueens(n, 0, colVisited, d1, d2);
        return count;
    }

    public void findQueens(int n, int row, boolean[] colVisited, boolean[] d1, boolean[] d2) {
        if (row == n) {
            //到末尾，则加
            ++count;
            return;
        }
        for (int j = 0; j < n; j++) {
            if (colVisited[j] || d1[row + j] || d2[row - j + n - 1]) {
                continue;
            }
            colVisited[j] = true;
            d1[row + j] = true;
            d2[row - j + n - 1] = true;
            findQueens(n, row + 1, colVisited, d1, d2);
            colVisited[j] = false;
            d1[row + j] = false;
            d2[row - j + n - 1] = false;
        }
        return;
    }

• 传入一个count数组，用作指针的作用

    public int totalNQueens(int n) {
        if (n <= 0) {
            return 0;
        }
        boolean[] d1 = new boolean[2 * n - 1];
        boolean[] d2 = new boolean[2 * n - 1];
        boolean[] colVisited = new boolean[n];
        //传入count数组，用count[0]来统计次数，类似于指针
        int[] count = new int[1];
        findQueens(n, 0, colVisited, d1, d2, count);
        return count[0];
    }

    public void findQueens(int n, int row, boolean[] colVisited, boolean[] d1, boolean[] d2, int[] count) {
        if (row == n) {
            ++count[0];
            return;
        }
        for (int j = 0; j < n; j++) {
            if (colVisited[j] || d1[row + j] || d2[row - j + n - 1]) {
                continue;
            }
            colVisited[j] = true;
            d1[row + j] = true;
            d2[row - j + n - 1] = true;
            findQueens(n, row + 1, colVisited, d1, d2, count);
            colVisited[j] = false;
            d1[row + j] = false;
            d2[row - j + n - 1] = false;
        }
        return;
    }

• 将dfs函数返回值改成int，返回

    public int totalNQueens(int n) {
        if (n <= 0) {
            return 0;
        }
        boolean[] d1 = new boolean[2 * n - 1];
        boolean[] d2 = new boolean[2 * n - 1];
        boolean[] colVisited = new boolean[n];
        //改变函数返回值，直接返回可能性结果
        int res = findQueens(n, 0, colVisited, d1, d2);
        return res;
    }

    public int findQueens(int n, int row, boolean[] colVisited, boolean[] d1, boolean[] d2) {
        if (row == n) {
            //符合，返回1
            return 1;
        }
        int res = 0;
        for (int j = 0; j < n; j++) {
            if (colVisited[j] || d1[row + j] || d2[row - j + n - 1]) {
                continue;
            }
            colVisited[j] = true;
            d1[row + j] = true;
            d2[row - j + n - 1] = true;
            //累加所有可能性结果
            res += findQueens(n, row + 1, colVisited, d1, d2);
            colVisited[j] = false;
            d1[row + j] = false;
            d2[row - j + n - 1] = false;
        }
        return res;
    }