n皇后方案数

#include"bits/stdc++.h"
using namespace std;
typedef long long LL;
LL ans;
//状压每一位表示每一列是否有放
//sta表示当前放了多少个
//ld是副对角线 ,rd是主对角线 
LL tot;//tot表示全部都放好了 
void dfs(LL sta,LL ld,LL rd)
{
    if(sta==tot)
    {
        ans++;
        return ;
    }
    LL pos=tot&(~(sta|ld|rd));         //取出还能放的位置
    while(pos)
    {
        LL t=pos&-pos;                  //lowbit取出每一个位置来放一放
        dfs(sta|t,(ld|t)<<1,(rd|t)>>1); //(ld|t)<<1 表示下一行t这个位置的左边就想当于已经放了东西,那么下次你就不能放了 
        pos-=t;
    } 
}
int main()
{
    LL N;
    while(cin>>N)
    {
        ans=0;
        tot=((1LL)<<N)-1;
        dfs(0,0,0);
        cout<<ans<<endl;
    }
}

改成有返回值的:

#include"bits/stdc++.h"
using namespace std;
typedef long long LL;
LL dfs(LL END,LL sta,LL ld,LL rd)
{
    if(sta==END)return 1;
    else if(sta>END)return 0;
    LL pos=(END)&(~(sta|ld|rd));
    LL ans=0; 
    while(pos)
    {
        LL t=pos&-pos;
        ans+=dfs(END,sta|t,(ld|t)<<1,(rd|t)>>1);
        pos-=t;
    }
    return ans;
}
int main()
{
    LL N;
    while(cin>>N)
    {
        cout<<dfs((1<<N)-1,0,0,0)<<endl;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值