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;
    }
}