基本N皇后问题(位运算优化)

最新推荐文章于 2022-03-19 12:31:54 发布

原创最新推荐文章于 2022-03-19 12:31:54 发布 · 418 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种使用位运算和递归方法解决N皇后问题的C++实现方案。该算法通过位运算快速判断皇后的位置冲突，并利用递归进行状态搜索，高效地找到所有可行解。

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

#include <iostream>
#include <vector>
#include <math.h>
using namespace std;
int uplimit;
int sum = 0;
int n;
vector<int> res;
void doans(int r, int ld, int rd)
{
    if (r != uplimit)
    {
        int pos=uplimit&~(r | ld | rd);
        while (pos)
        {
            int p = pos&-pos;
            pos -= p;
            res.push_back(n-log2(p));
            doans(r | p, (ld | p) << 1, (rd | p) >> 1);
            res.pop_back();
        }
    }
    else
    {
        for (int i = 0; i < res.size(); ++i)
            cout << res[i] << " ";
        cout << endl;
        sum++;
    }
}
int main()
{
    while (cin >> n)
    {
        if (n == 0)
            break;
        sum = 0;
        res.clear();
        uplimit = (1 << n) - 1;
        doans(0, 0, 0);
        cout << sum << endl;
    }
    return 0;
}