libreoj#10170. 「一本通 5.4 例 1」国王

libreoj
题解：状压dp，定义dp[i][j][k]为：前i行以状态s[j]为结尾填进去k个国王的情况数

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int inf=2e9;
const LL mod=10007;
LL dp[11][205][205];
int N,K;
int s0,con[205],s[205];
void work1()
{
	s0=0;
	for(int i=0;i<(1<<N);i++){
		if(i&(i<<1))continue;
		int k=0;
		for(int j=0;j<N;j++){
			if((i>>j)&1)
				k++;
		}
	    s[++s0]=i;
	    con[s0]=k;
	}
}
void work2()
{
	dp[0][1][0]=1;
	for(int i=1;i<=N;i++){
		for(int j=1;j<=s0;j++){
			for(int k=0;k<=K;k++){
				if(k>=con[j]){
					for(int x=1;x<=s0;x++){
						if(!(s[x]&s[j])&&!(s[x]&(s[j]<<1))&&!(s[x]&(s[j]>>1))){
							dp[i][j][k]+=dp[i-1][x][k-con[j]];
						}
					}
				}
			}
		}
	}
	LL res=0;
	for(int i=1;i<=s0;i++){
		res+=dp[N][i][K];
	}
	printf("%lld\n",res);
}
int main()
{
	cin>>N>>K;
	work1();
	work2();
    return 0;
}