状压dp SCOI 2005 互不侵犯

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

考虑如何状压，对于每一行，我们用一个长度为$n$的二进制串表示每一行的状态，比如$1010$表示第一个，第三个位置上有国王，而$1010$对应的十进制数为$10$，我们将每一行的状态都用二进制表示然后状压成一个十进制数。一共有$2^n-1$种状态，因为最多的就是一行全为$1$的情况。

接下来设计状态：$dp[i][j][k]$表示第$i$行的状态为$j$（$j$即为状压后的十进制数），前$i$行一共使用了$k$个国王。这里定义$num[i]$表示一行状态为$i$时的国王个数，$k$为当前这一行的状态，$j$为上一行的状态，$q$为到前一行总共放的国王个数。所以：$$dp[i][k][q+num[k]]+=dp[i-1][j][q];$$

首先我们先预处理出对于一行内哪些状态是合法的，如果状态$i$合法即记$book[i]=1$，对于每一种合法状态，我们处理出合法这种状态需要的国王个数$num[i]$。对于同一行，一个国王不能有左右相邻的国王，所以对于状态$i$，我们让$i$&$(i>>1)，i$&$(i<<1)$如果二者都为$0$，即为合法。对于一个合法状态$i$，我们对它进行二进制分解，求出其中有多少个$1$，即为$num[i]$，然后设出初始状态$f[1][x][num[x]]=1$。

for(ll i=0;i<(1<<n);++i)
{
	if(!(i&(i>>1))&&!(i&(i<<1)))
		book[i]=1;
	ll w=i;
	while(w)
	{
		if(w%2==1)
			num[i]++;
		w/=2;	
	}
	if(book[i])	
		dp[1][i][num[i]]=1;
}

接下来我们分别枚举每一行$i$，前一行的状态$j$，如果前一行的状态合法，就继续枚举当前行的所有状态$k$，对于当前行的所有状态，我们首先判断它是否合法，然后让前一行的状态分别&$k，(k<<1)，(k>>1)$，如果均为$0$即视为合法，枚举到前一行总共放的国王个数根据方程进行累加转移。

最后枚举最后一行的所有状态累加一下答案即可。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll dp[20][1000][100],num[1000];
bool book[1000];
ll n,king,ans;
int main()
{
	cin>>n>>king;
	for(ll i=0;i<(1<<n);++i)
	{
		if(!(i&(i>>1))&&!(i&(i<<1)))
			book[i]=1;
		ll w=i;
		while(w)
		{
			if(w%2==1)
				num[i]++;
			w/=2;	
		}
		if(book[i])	
			dp[1][i][num[i]]=1;
	}
	for(ll i=2;i<=n;++i)
		for(ll j=0;j<(1<<n);++j)
			if(book[j])
				for(ll k=0;k<(1<<n);++k)
					if(book[k]&&!(k&j)&&!((k<<1)&j)&&!((k>>1)&j))
						for(ll q=num[j];q+num[k]<=king;++q)
							dp[i][k][q+num[k]]+=dp[i-1][j][q];
	for(ll i=0;i<(1<<n);++i)
		ans+=dp[n][i][king];
	cout<<ans;
	return 0;
}