题目链接:https://www.luogu.org/problemnew/show/P1896
记:dp[i][j][k]为前i行,当前状态序列号为j,共放置的国王数为k的方法数。
显然:dp[i][j][k]=sum(dp[i-1][s][k-num]j])(枚举上一行s并且判断合法性即可)
其中:s&j==0同时s&j<<1==0同时s<<1&j==0(黑科技)
即上一行的s状态与当前行j状态的国王之间不存在冲突。(QWQ)
///!!!!!!!本题坑点:LONG LONG INT 统计dp和ans,否则会wa3个点。
AC代码:
#include<cstdio>
#include<iostream>
using namespace std;int sum=0;int v[1024];int num[1024];
long long int dp[17][207][107];
void find(int x){
int f=1,ans=0;while(f<=v[x]){
if((f&v[x])!=0){
ans++;
}
f=f<<1;
}num[x]=ans;
}void ycl(int c){
int f=0;while(f<=(1<<c)-1){
if((f&(f<<1))==0){
sum++;v[sum]=f;find(sum);
}f++;
}
}
int main(){
int i,j,n,k;scanf("%d%d",&n,&k);
ycl(n);for(i=1;i<=sum;i++){//第一行单独处理
dp[1][i][num[i]]=1;
}/*for(i=1;i<=sum;i++)printf("%d ",num[i]);printf("\n\n");
for(int g=1;g<=sum;g++){
for(int h=0;h<=k;h++){
printf("%d ",dp[1][g][h]);
}printf("\n");
}printf("\n");*/
for(i=2;i<=n;i++){
for(j=1;j<=sum;j++){
for(int w=num[j];w<=k;w++){
for(int s=1;s<=sum;s++){
if((v[s]&v[j])==0&&(v[s]&(v[j]<<1))==0&&(v[j]&(v[s]<<1))==0){
dp[i][j][w]+=dp[i-1][s][w-num[j]];//if(i==2&&j==2&&w==1&&s==4)printf("!!!\n");
}
}
}
}
/*for(int g=1;g<=sum;g++){
for(int h=0;h<=k;h++){
printf("%d ",dp[i][g][h]);
}printf("\n");
}printf("\n");*/
}long long int ans=0;for(i=1;i<=sum;i++){
ans+=dp[n][i][k];
}printf("%lld",ans);
}