[BZOJ]1087: [SCOI2005]互不侵犯King 状态压缩DP

Description

　　在N×N的棋盘里面放K个国王，使他们互不攻击，共有多少种摆放方案。国王能攻击到它上下左右，以及左上
左下右上右下八个方向上附近的各一个格子，共8个格子。

Input

　　只有一行，包含两个数N，K （ 1 <=N <=9, 0 <= K <= N * N）

Output

　　方案数。

Sample Input

3 2

Sample Output

16

这种题目一看就知道是状态压缩DP，随便搞就是了。

代码：

#include<cstdio>
#include<cstring>
long long ans=0,f[11][515][83];//f[i][j][k]表示i行状态为j，已经放了k个国王的方案数 
int one[515],list[515],len=0;
int onenum(int n)
{
    int r=0;
    while(n!=0)
	{
        n=n&(n-1);
        r++;
    }
    return r;
}
bool check1(int x)
{
	if((x&(x<<1))==0) return true;
	return false;
}
bool check2(int x,int y)
{
	if((x&y)!=0) return false;
	if((x&(y<<1))!=0) return false;
	if((x&(y>>1))!=0) return false;
	return true;
}
int main()
{
	memset(f,0,sizeof(f));
	int n,k;
	scanf("%d%d",&n,&k);
	for(int i=0;i<(1<<n);i++)
	if(check1(i)==true)
	{
		one[i]=onenum(i);
		if(one[i]<=k)
		{
			list[++len]=i;
			f[1][i][one[i]]=1;
		}
	}
	for(int i=2;i<=n;i++)
	{
		for(int o=0;o<=k;o++)
		{
			for(int j=1;j<=len;j++)
			{
				int x=list[j];
				if(one[x]>o) continue;
				for(int k=1;k<=len;k++)
				{
					int y=list[k];
					if(one[x]+one[y]>o) continue;
					if(check2(x,y)==false) continue;
					f[i][x][o]+=f[i-1][y][o-one[x]];
				}
			}
		}
	}
	for(int i=0;i<(1<<n);i++) ans+=f[n][i][k];
	printf("%lld",ans);
}