BZOJ 1087: [SCOI2005]互不侵犯

Description

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

Input

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

Output

　　方案数。

Sample Input

3 2

Sample Output

16

 

题目链接(BZOJ)

数据(LOJ)

 

首先看到数据范围第一反应这题不是用搜索就是状压dp。搜索我只会爆搜打表，好像并没有什么高级搜索技巧能做。

所以状压dp是最好的选择(实际这道题就是状压dp的入门题)。

状态压缩，顾名思义就是把状态通过2进制的方式存起来。在这道题中，对于每行棋盘的n个格子，0代表不放国王，1代表放国王。打个比方，如果n=3,111代表3个格子都放了国王(本题中这种情况不合理)，101表示第一个格子和第三个格子放了国王。

很显然每种方案只会对应一个唯一的十进制的数，所以可以通过枚举方案来避免复杂度。

接下来是dp数组的建立。首先，状压dp有一维肯定表示状态，然后再用一维表示行数，最后还要用一维表示已经放置的国王数。这个数组的值便是在第行已经放了个国王且这一行状态为时的方案数。

然后就是如何判断状态可行性的问题了。因为是从第一行往下枚举，所以第一行只需要判断自己这一行是否合法。因为国王能攻击左边和右边，所以对于状态来说，每个二进制下的1的左右必定都是0，所以只需要将左移/右移再和取个并。也就是只有当(k<<1&k)==0&&(k>>1&1)==0时状态才合法。

对于之后的每行来说，除了刚才那种判断，还要判断是否和上一行状态冲突。由于国王能攻击左上，上，右上。假设当前行状态为，上一行状态为，除了(k<<1&k)==0&&(k>>1&1)==0之外，还必须满足(k<<1&j)==0&&(k>>1&j)==0&&(k&j)==0)。

所以在预处理好第一行之后，

枚举剩下的行数，当前行状态，上一行状态，放完当前行后已经放置的国王数。然后转移就好了，转移方程就直接看代码吧。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<set>
#include<cmath>
#include<cstring>
#include<queue>
using namespace std;
inline int read()
{
    int sum=0,t=1;
    char ch =getchar();
    while(ch<'0'||ch>'9')
	{
		if(ch=='-') t=-1;
        ch=getchar();
	}
    while(ch>='0'&&ch<='9')
    {
        sum=sum*10+ch-'0';
        ch=getchar();
    }
    return sum*t;
}
void write(int x)
{
	if(x>9) write(x/10);
	putchar(x%10+'0');
}
inline int getbit(int x)//虽然有O(1)的方法但是既然复杂度过得去还是O(logx)暴力比较方便
{
	int sum=0;
	while(x)
	{
		sum+=(x&1);
		x>>=1;
	}
	return sum;
}
int n,k;
long long dp[11][105][1100];
long long ans=0;
int main()
{
	n=read();k=read();
	int maxn=(1<<n)-1;
	for(register int i=0;i<=maxn;++i)//预处理第一行
		if((i<<1&i)==0&&(i>>1&i)==0)
			dp[1][getbit(i)][i]=1;
	for(register int i=2;i<=n;++i)//枚举行数
		for(register int j=0;j<=maxn;++j)//上一行状态
			if((j<<1&j)==0&&(j>>1&j)==0)
				for(register int p=0;p<=maxn;++p)//当前行状态
					if((p<<1&p)==0&&(p>>1&p)==0&&(p<<1&j)==0&&(p>>1&j)==0&&(p&j)==0)
						for(register int bt=getbit(p),q=bt;q<=k;++q)//已经放置的国王数
							dp[i][q][p]+=dp[i-1][q-bt][j];//已经放置的国王数-当前行放置的国王数
	for(register int i=0;i<=maxn;++i)                     //=上一行状态已经放置的国王数
		ans+=1LL*dp[n][k][i];
	cout<<ans<<endl;
}