和前面的玉米田很接近,加了数量的要求。做的时候翻了两个ruozhi错误,一是估计数据范围很大,但没用longlong.二是没有仔细看k的范围,数组f的第三维开小了。
#include<iostream>
using namespace std;
const int maxn=512;
int n,K,num,c[maxn];
long long ans=0;
void count(){
for(int i=0;i<512;i++){
int j=i;
while(j>0){ c[i]+=(j%2);j/=2;}
}
}
long long f[10][512][82];
long long dp(){
f[0][0][0]=1;
num=(1<<n);
for(int k=1;k<=K;k++)
for(int i=1;i<=n;i++){//处理每行
for(int j=0;j<num;j++){
if(j&(j<<1))continue;//左边相邻,不做。
if(k-c[j]<0)continue;//放多了,不做
for(int l=0;l<num;l++){
if((l&j)||((l<<1)&j)||((l>>1)&j))continue;//与上一行冲突
f[i][j][k]+=f[i-1][l][k-c[j]];
}
}
}
for(int i=1;i<=n;i++)
for(int j=0;j<num;j++)ans+=f[i][j][K];
return ans;
}
int main(){
cin>>n>>K;
count();
cout<<dp();
}