题目戳这里: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.