整道题做的非常艰难,最后一直纠结s=0能ac,s=1不能ac的问题。其实s=0这块没毛病,主要是最后计算ans的时候,只加上最后一行国王为k的所有状态方案之和。
#include<bits/stdc++.h>
using namespace std;
int n,k;
long long dp[10][1<<9+1][82];
int state[1<<9+1];
int c[1<<9+1];
int cnt=0;
//这道题当且仅当比较上下行是否冲突采用state数组
//否则其他都用cnt来代表状态,更为简便
void init(){//进行预处理,获得每个合法的行状态,并且记录该状态的行个数
for(int i=0;i<=(1<<n)-1;i++){//枚举所有状态
if(i&(i<<1))continue;
state[++cnt]=i;
int t=i;
while(t){//获取1的个数,代表国王个数
c[cnt]+=t%2;
t>>=1;
}
}
}
int main(){
long long ans=0;
cin>>n>>k;
init();
for(int i=1;i<=cnt;i++)//利用刚刚存储的信息更新第一行
if(c[i]<=k)dp[1][i][c[i]]=1;//第一行的每种可行状态就是一种方案
//这里的i表示第i种状态,因为前面已经将所有状态预处理出来了
for(int i=2;i<=n;i++)//枚举接下来的n-2行
for(int j=1;j<=cnt;j++)//对于每一行,枚举这一行和上一行的cnt种状态
//他们的状态分别用j,p来表示
{
for(int p=1;p<=cnt;p++)
{
if(state[j]&state[p])continue;//正对
if((state[j]<<1)&state[p])continue;//左上角
if((state[j]>>1)&state[p])continue;//右上角
//假设目前的行以上的国王个数为s
for(int s=0;s<=k-c[j];s++)
dp[i][j][c[j]+s]+=dp[i-1][p][s];
}
}
//
for(int i=1;i<=cnt;i++)
ans+=dp[n][i][k];
cout<<ans;
system("pause");
return 0;
}