leetcode第一刷_N-Queens II

最新推荐文章于 2024-08-25 06:11:42 发布

Mr_Giraffe

最新推荐文章于 2024-08-25 06:11:42 发布

阅读量1.2k

点赞数

分类专栏： leetcode 算法文章标签： c++ leetcode 算法

本文链接：https://blog.csdn.net/u012792219/article/details/25701785

版权

算法同时被 2 个专栏收录

158 篇文章 0 订阅

订阅专栏

leetcode

143 篇文章 0 订阅

订阅专栏

这个题好无趣，居然输出解的个数，前一个题把所有解都输出出来了，还愁不知道解的个数吗。。

我怀疑这个解的个数是有一个类似通项的东西，就上网查了一下，没有啊亲，最后就把上一题的代码稍微改了一下过掉了。

class Solution {
public:
    int totalNQueens(int n) {
        if(n == 0)  return 0;
        int pos[n], res=0, start=0;
        bool flag, vis[n];
        memset(pos, 0, sizeof(pos));
        memset(vis, 0, sizeof(vis));
        int i=0, j;
        while(i<n){
            for(j=start;j<n;j++){
                if(vis[j])  continue;
                flag = true;
                for(int k=0;k<i;k++){
                    if(abs(k-i) == abs(pos[k]-j)){
                        flag = false;
                        break;
                    }
                }
                if(!flag) continue;
                vis[j] = 1;
                pos[i] = j;
                break;
            }
            if(j == n){
                --i;
                vis[pos[i]] = 0;
                start = pos[i]+1;
                if(i==0&&start>=n)  break;
                continue;
            }else{
                i++;
                start = 0;
            }
            if(i == n){
                res++;
                --i;
                vis[pos[i]] = 0;
                start = pos[i]+1;
                if(i==0&&start>=n)  break;
            }
        }
        return res;
    }
};