思路还是比较简单的,首先我们看到只有N<=9的数据范围,过多的不考虑,直接状压dp莽一波,每一次dfs的时候记录下上一行的状态然后不断的枚举这一层的状态,对于每一个状态根据题意再写一个check函数就ok了,可是时间范围还是很难满足,加上记忆化
1.不过需要注意的是,这里的记忆化定义,由于还受到了放置国王个数的限制所以需要再开一维。
2.在一开始写check函数的时候,把upcnt写成了last也是郁闷,所以写代码的时候一定要思路清晰,搞清楚各个变量名
#include<cstdio>
#include<cstring>
#include<iostream>
#define ll long long
using namespace std;
ll n,k,ans,f[12][6000][100];//f[i][j]表示在第i层 当前状态数是j能够符合条件的方案数
ll bitcount(ll x){return x==0?0:bitcount(x/2)+(x&1);}
bool check(ll pos,ll last,ll upcnt){//last是指上一层的状态 upcnt是指上面所有的国王数
ll x=bitcount(pos);
if(x+upcnt>k)return false;
for(ll i=1;i<=n;i++)if((1<<i-1)&pos){
if(i>1)if((last&(1<<i-2))||(pos&(1<<i-2)))return false;
if(i<n)if((last&(1<<i))||(pos&(1<<i)))return false;
if(last&(1<<i-1))return false;
}
return true;
}
ll dfs(ll o,ll last,ll upcnt){
if(upcnt>k)return 0;
if(f[o][last][upcnt]!=-1)return f[o][last][upcnt];
if(o>n)return upcnt==k?1:0;
f[o][last][upcnt]=0;
for(ll i=0;i<(1<<n);i++){
if(check(i,last,upcnt))f[o][last][upcnt]+=dfs(o+1,i,upcnt+bitcount(i));
}
return f[o][last][upcnt];
}
int main(){
scanf("%lld%lld",&n,&k);
memset(f,-1,sizeof(f));
for(ll i=0;i<(1<<n);i++){
if(check(i,0,0))ans+=dfs(2,i,bitcount(i));
}
printf("%lld",ans);
return 0;
}