[LeetCode] N-Queens II

zjjee

于 2014-12-03 15:33:57 发布

阅读量256

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode

本文链接：https://blog.youkuaiyun.com/zjjee/article/details/41698289

LeetCode 专栏收录该内容

131 篇文章

订阅专栏

本文介绍了一种解决N皇后问题的方法，该方法不仅找出所有可能的解决方案配置，还计算了这些配置的总数。通过递归算法和冲突检查机制实现了这一目标。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Follow up for N-Queens problem.

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

class Solution {
public:
    int cnt = 0;
    int *col;
    int totalNQueens(int n) {
        col = new int[n];
        generate(n,0);
        return cnt;
    }
    void generate(int n,int dep){
        if(dep == n){
            cnt ++;
            return;
        }
        for(int i = 0;i < n;i ++){
            if(check(dep,i)){
                generate(n,dep + 1);
                col[dep] = -1;
            }
        }
    }
    bool check(int k,int i){
        col[k] = i;
        for(int i = 0;i < k;i ++)
            if(col[i] == col[k] || abs(col[i] - col[k]) == abs(i - k)) return false;
        return true;
    }
};