SCOI2005[互不侵犯KING]

[SCOI2005]互不侵犯King

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 2415   Solved: 1421
[ 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

Source   

-这么小的数据，难道是搜索？

-不过考虑每个格子的01状态，就有2^81，貌似不行呀。

-那就状压吧！！

-不会状压啊？

-我也是生手~~

我写过的状压有 和图论连在一起的，有选择性的，还有经典的铺砖。

一开始尝试像铺砖那样，构造每一层和上一层，然后通过上一层转移。

不过这道题可不行，它有没说每层棋子要放满- -

不过略加分析就可以知道它的每一层状态都可以由前一层转移过来。

我们定义f[i][j][k]表示当前第i层，状态为j,选了k个棋子的方案数。

f[i][j][k]=∑ f[i-1][p][k-g(p)]       p是合法的上一层状态，g(p)是p的棋子数。

考虑每一次枚举，状态有2^9种=512,512*512*9*9*9 >1亿。(枚举2层状态，枚举行数，枚举更新棋子数，检查合法)。

貌似会超时吧！

所以我们提前处理出可能状态。

每一层对于自己i需要没有两个相邻的1,逐位检查或者运用为运算的技巧i&(i>>1)==0&&i&(i<<1)==0.

上下两层也要满足的话，由于状态数512,512^2的数组还是开的下。

检查差不多，诸位比较，循环次数9.或者位运算i&x==0&&i&(x>>1)==0&&i&(x<<1)==0.

然后就巴拉巴拉写代码了：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<queue>
#include<cmath>
#define LL long long
using namespace std;
LL n,k;
LL f[10][1<<11][90];//f[i][j][k]表示前i行  当前行状态j  棋子有k个 
LL gp[1<<11];//每个状态有几个棋子
LL sum;
bool map[1<<11][1<<11];//当前状态和上一层状态是否合法 
LL det[1<<11];
LL cnt=0;
void init()
{
	for(LL i=0;i<=sum;i++)
	{		
		if(!(i&(i<<1))&&!(i&(i>>1)))
		{
			det[++cnt]=i;
			for(LL j=0;j<n;j++)
		    {
			    if(i&(1<<j))gp[cnt]++;
		    }
		}
		
	} 
	for(LL i=1;i<=cnt;i++)
	for(LL j=i;j<=cnt;j++)
	{
		
		bool ok=true;
		if(det[i]&det[j])ok=false;
		if(det[i]&(det[j]<<1))ok=false;
		if(det[i]&(det[j]>>1))ok=false;
		if(ok)map[i][j]=map[j][i]=true;
	}
}
int main()
{
	scanf("%lld%lld",&n,&k);
	sum=(1<<n)-1;
	init();
	f[0][1][0]=1;
	for(LL i=0;i<n;i++)
	{
		for(LL j=1;j<=cnt;j++)
		{
			for(LL l=0;l<=k;l++)
			{
				if(f[i][j][l])
				{
					for(LL p=1;p<=cnt;p++)
					{
						if(map[j][p]&&l+gp[p]<=k)
						{
							f[i+1][p][l+gp[p]]+=f[i][j][l];
						}
					}
				}
			}
		}
	}
	LL ans=0;
	for(LL i=1;i<=sum;i++)ans+=f[n][i][k];
	printf("%lld\n",ans);
	return 0;
}