BZOJ1087 [SCOI2005]互不侵犯King 状压DP

#include<stdio.h>
#include<algorithm>
#include<queue>
#include<string.h>
#include<math.h>
#include<set>
#include<map>
#include<vector>
#include<iomanip>
#include<stack>
using namespace std;
#define ll long long
#define ull unsigned long long
#define pb push_back
#define mem(a) memset(a,0,sizeof a)
#define FOR(a) for(int i=1;i<=a;i++)
#define sqr(a) (a)*(a)

const int maxn=6e2+5;	//1<<9

int n,k;
ll ans;
bool s[maxn],G[maxn][maxn];	//判状态是否合法，判能否邻行
int num[maxn];				//状态i有多少个1位
ll f[10][100][600];			//前i行放j个1，第i行状态为k
int sum;					//sum种状态

void pre_s(){
for(int i=0;i<sum;i++){
if((i&(i<<1))==0){	//本行没相邻
s[i]=true;
int t=i,cnt=0;
while(t){
cnt+=(t&1);
t>>=1;
}
num[i]=cnt;
f[1][cnt][i]=1;
}
}
}
void pre_G(){
for(int i=0;i<sum;i++){
if(!s[i])continue;
for(int j=0;j<sum;j++){
if(!s[j])continue;
if( (!(i&j)) && (!((i<<1)&j)) && (!((i>>1)&j)) )G[i][j]=true;
}
}
}

void dp(){
for(int i=2;i<=n;i++)
for(int j=0;j<=k;j++)
for(int now=0;now<sum;now++){
if(!s[now])continue;
if(num[now]>j)continue;
for(int fr=0;fr<sum;fr++){
if(!s[fr])continue;
if(!G[fr][now])continue;
if(num[fr]+num[now]>j)continue;
f[i][j][now]+=f[i-1][j-num[now]][fr];
}
}
}

int main(){
scanf("%d%d",&n,&k);
sum=1<<n;
pre_s();
pre_G();
dp();
for(int i=0;i<sum;i++){
if(!s[i])continue;
ans+=f
[k][i];
}
printf("%lld\n",ans);
}