bzoj1087状压dp

题目戳这里：http://www.lydsy.com/JudgeOnline/problem.php?id=1087
这是一道简单的状压dp。
我们首先预处理一行中满足要求的状态，同时记录每个合法状态的国王king的个数。然后预处理每两行满足要求的状态，然后跑一遍4重循环即可。
用dp[i][j][k]表示第i行j状态k个国王最大方案数。
dp[i][j][k]+=dp[i-1][p][k-cnt[j]]
其中p是枚举的状态，cnt[j]表示一行中j状态下有多少国王。
代码如下：
简单易懂

#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 600
using namespace std;
int n,k,zt,s[N],tot,cnt[N];
long long dp[15][N][100];
int c[N][N];
bool ok(int x){
    int pia=0,sum=0;
    for(int k=x;k;k>>=1)
    if(k&1){
        sum++;
        pia++;
        if(pia==2)return false;
    }
    else pia=0;
    cnt[x]=sum;
    return true;
}
bool judge(int z1,int z2){//z1 up z2 down
    for(int p=0;z1;z1>>=1,p++){
        if(z1&1){
            if(((z2>>p)&1)||((z2>>(p+1))&1))return false;
            if(p>0&&(z2>>(p-1))&1)return false;
        }
    }
    return true;
}
void init(){
    for(int i=0;i<=zt;i++)
    if(ok(i)==true) s[++tot]=i;
    for(int i=1;i<=tot;i++)
    for(int j=1;j<=tot;j++)
    if(judge(s[i],s[j])==true)c[i][j]=1;
}
int main(){
    scanf("%d%d",&n,&k);
    zt=(1<<n)-1;
    init();
    for(int i=1;i<=tot;i++)
    dp[1][s[i]][cnt[s[i]]]=1;
    for(int i=2;i<=n;i++){
        for(int j=1;j<=tot;j++){
            for(int q=1;q<=tot;q++){
                if(c[j][q])
                for(int p=0;p<=k;p++)
                if(p-cnt[s[j]]>=0)
                dp[i][s[j]][p]+=dp[i-1][s[q]][p-cnt[s[j]]];
            }
        }
    }
    long long ans=0;
    for(int i=1;i<=tot;i++)
    ans+=dp[n][s[i]][k];
    printf("%lld\n",ans);
    return 0;
}

记得开long long，这个我忘开调了好久0 0.