N皇后-二进制剪枝

题目描述
n皇后问题:一个n×n的棋盘,在棋盘上摆n个皇后,满足任意两个皇后不能在同一行、同一列或同一斜线上的>方案有多少种?

我们知道一个N*N的棋盘,如果标记皇后所在位置为1,不在的位置为0,那么用二进制表示最为节省空间。

#include <iostream>
using namespace std;
// ================= 代码实现开始 =================
//ans:总答案
//allOne:用于二进制&的全1数
int ans, allOne;
//搜索(用二进制来优化)
//pos:其二进制上的某个位置的1表示当前所在行的相应的位置(列)已经放了一个皇后
//left:其二进制上的某个位置的1表示当前所在行的相应的位置(是由于右对角线上已经有皇后),不能放置皇后
//right:其二进制上的某个位置的1表示当前所在行的相应的位置(是由于左对角线上已经优皇后),不能放置皇后
void dfs(int pos,int left, int right){
    if(pos == allOne){//当且仅当每一列都放了一个皇后,那么整个棋盘已经放了n个合法皇后,故要终止
        ++ans;
        return;
    }
    for(int t = allOne & (~(pos | left | right)); t>0;){//t代表可以放的集合对应的二进制数
        int p = t & -t;//low bit: 二进制中最右边的1
        dfs(pos | p, ((left | p) << 1) & allOne, (right | p) >> 1);
        t ^= p;
    }
}
//一个n*n的棋盘,在棋盘上摆n个皇后,求满足任意两个皇后不能在同一行,同一列或同一斜线上的方案数
//n:上述n
//返回值:方案数
int getAnswer(int n){
    ans = 0;
    allOne = (1 << n) - 1;
    dfs(0, 0, 0);
    return ans;
}
int main() {
    int n;
    cin >> n;
    cout << getAnswer(n) << endl;
    return 0;
}

其中所运用到的二进制使用方法

  • lowbit 取出t中最后一个1
int p = n & (-n);

其原理是负数的补码为反码加上1,加上的1会落在第一个0空位上,这个0空位会恢复到原来的1,所以取出第一个1

  • allone = (1<<n) -1
    可以知道1左移n位之后在减去1 得到n个1 用来取出int 所需要的位数 由n决定。
    那么在本体中 pos | left | right 为一行中已经放下的位置 那么没放下的位置都是0 怎么取出空位?
    只要~(取反) 然后 用allone &(相与) 再 用lowbit 取出最后一个1 就是可以摆放的位置,进行搜索。
    和allone 取 & 本来0还是0 本来1还是1 在格子右移出棋盘的时候进行截断。
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 网站乱码问题我们会经常碰到,大多见于非英文的中文字符或其他字符乱码,而且,这类问题常常是因为编码方式问题,主要原因...
    波段顶底阅读 2,962评论 1 9
  • 离开的第二天 回学校的第一天 想他的不知道多少天 跟同学一起聊天话题都是他 无聊的时候看他的照片 抱抱卡姐无意识亲...
    uaremybelief阅读 207评论 0 0
  • “有些鸟儿是注定关不住的,因为它们的每一片羽毛,都沾满了自由的光辉” 我赞美候鸟,赞美你的境界高远,追求自由的胸襟...
    一瓢耳阅读 1,269评论 0 7
  • 万事转头空 哀思几万重 青山埋忠骨 云岭血染红
    胡桃木_f625阅读 380评论 0 1