BZOJ 1087-互不侵犯King（状压DP）

1087: [SCOI2005]互不侵犯King

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 3984   Solved: 2323
[ Submit][ Status][ Discuss]

Description

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

Input

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

Output

　　方案数。

Sample Input

3 2

Sample Output

16

HINT

DP[ i ][ j ][ k ];表示到第i行已经放了j个国王，且第i行状态为k的方案数

#include<map>
#include<queue>
#include<math.h>
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
typedef long long  ll;
ll dp[10][100][520],flag[520][520],cnt[520],n,k,ans,mark[520];
int  main(void)
{
	ll i,j,tmp,k,h,m;
	scanf("%lld%lld",&n,&m);
	for(i=0;i<(1<<n);i++)
	{
		if(i&(i>>1))
			continue;
		tmp=0;
		for(j=i;j>0;j>>=1)
			tmp+=(j&1);
		cnt[i]=tmp;
		mark[i]=1;
	}
	for(i=0;i<(1<<n);i++)
		for(j=0;j<(1<<n);j++)
			if(!(i&j) && !(i&(j>>1)) && !(j&(i>>1)))
				flag[i][j]=1;
	for(i=0;i<(1<<n);i++)
		if(mark[i])
			dp[1][cnt[i]][i]=1;
	for(i=1;i<n;i++)
		for(j=0;j<(1<<n);j++)
		{
			if(mark[j]==0)
				continue;
			for(k=0;k<(1<<n);k++)
			{
				if(mark[k]==0)
					continue;
				if(flag[j][k]==0)
					continue;
				for(h=cnt[j];h+cnt[k]<=m;h++)
					dp[i+1][cnt[k]+h][k]+=dp[i][h][j];
			}
		}
	for(i=0;i<(1<<n);i++)
		ans+=dp[n][m][i];
	printf("%lld\n",ans);
	return 0;
}