[bzoj1087]互不侵犯king

状压

• 代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
ll f[10][1030][50];
int tot,l[20],p[20],cnt,n,k;

bool legal(int st1){
    for(int i=1;i<=n;i++) l[i]=(st1>>(i-1))&1;
    for(int i=1;i<=n;i++) if(l[i]) 
        if(l[i+1]||l[i-1])return false;
    return true;
}
bool legal2(int st1,int stp){
    cnt=0;
    for(int i=1;i<=n;i++)
        l[i]=(st1>>(i-1))&1,p[i]=(stp>>(i-1))&1;
    for(int i=1;i<=n;i++) if(l[i]){
        cnt++;
        if(p[i]||p[i-1]||p[i+1]||l[i+1]||l[i-1])return false;
    }
    return true;
}
int main(){
    scanf("%d%d",&n,&k);
    if(k>25){printf("0\n");return 0;}
    tot=(1<<n)-1;
    f[0][0][0]=1;
    for(int i=1;i<=n;++i)
        for(int fromst=0;fromst<=tot;fromst++)if(legal(fromst))
        for(int tost=0;tost<=tot;tost++)if(legal2(tost,fromst))
            for(int j=0;j<=k;j++)if(j+cnt<=k)
                f[i][tost][j+cnt]+=f[i-1][fromst][j];
    ll ans=0;
    for(int i=0;i<=tot;i++) ans+=f[n][i][k];
    printf("%lld\n",ans);
}