【bzoj 1087】[SCOI2005]互不侵犯King 状压dp

思路还是比较简单的，首先我们看到只有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;
}