LeetCode 52. N-Queens II

最新推荐文章于 2021-01-22 11:28:53 发布

原创最新推荐文章于 2021-01-22 11:28:53 发布 · 347 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode

LeetCode 专栏收录该内容

350 篇文章

订阅专栏

本文介绍了一种简化版的N皇后问题解决方案，通过修改LeetCode 51.N皇后问题的实现，来计算所有可能的不冲突配置数量。代码采用C++编写，并实现了深度优先搜索算法。

Follow up for N-Queens problem.
Now, instead outputting board configurations, return the total number of distinct solutions.
分析：和LeetCode 51. N-Queens一样，只需改动几行代码即可~

class Solution {
    int cnt = 0;
public:
    int totalNQueens(int n) {
        vector<int> pos(n);
        dfs(pos, n, 0);
        return cnt;
    }
private:
    void dfs(vector<int> &pos, int n, int row) {
        if(row == n) {
            cnt++;
            return ;
        }
        for(pos[row] = 0; pos[row] < n; pos[row]++) {
            if(issafe(pos, n, row)) {
                dfs(pos, n, row + 1);
            }
        }
    }
    
    bool issafe(vector<int> &pos, int n, int row) {
        for(int i = 0; i < row; i++)
            if(pos[i] == pos[row] || abs(i - row) == abs(pos[i] - pos[row]))
                return false;
        return true;
    }
};