[LEETCODE]52. N-Queens II

[LEETCODE]52. N-Queens II

Follow up for N-Queens problem.

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

这道题比N-Queens 要简单一点。
它要求输出的所有可能的不同解。这里代码都是差不多的。

class Solution {public:    bool check(vector<int>& cols, int cur) {        for (int i = 0; i < cur; i++) {            if (cols[i] == cols[cur] || (abs(cols[i] - cols[cur]) == cur - i)) return false;        }        return true;    }    void Nqueens(int n, vector<int>& cols, int &res, int cur) {        if (cur == n) {            res++;            return;        }        for (int i = 0; i < n; i++) {            cols[cur] = i;            if (check(cols, cur)) {                Nqueens(n, cols, res, cur + 1);            }        }    }    int totalNQueens(int n) {        vector<int> cols(n, -1);        int res = 0;        Nqueens(n, cols, res, 0);        return res;    }};

主要就是用了一个递归回溯，要注意递归的话一定要设置一个返回条件，然后用check 来判断是否符合要求。