互不侵犯（状压dp）

再来跟大家分享一道状压dp的问题，在这道问题中我们能学到一些处理限制条件的位运算的技巧。

题目描述

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

注：数据有加强（2018/4/25）

输入格式

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

输出格式

所得的方案数

输入输出样例

输入 #1

3 2

输出 #1

16

分析：这是一道经典的状压dp问题，首先我们来分析一下dp数组的状态，一般处理这种棋盘问题我们都会有两维数组，一组表示当前处理的行数，另一维表示当前行的状态，但是对于这道题而言只用这两维好像不能完全表示清楚当前状态，比如当前已经使用的几个国王，说到这，相信大家就不难想出这道题是利用三维dp数组来进行状态转移，先来说一下dp含义，dp[i][j][k]表示第i行的国王摆放状态是j，且第i行及之前已经用去了k个国王，下面我来说一下如何处理限制条件，

假如我们处理的当前行状态为i，上一行状态为j，那么

（i&j）==0  说明相邻两行上下无相邻的1

（ i & ( i >> 1 ) ）==0说明当前行无相邻的1

( (i>>1) & j )==0 说明当前行左下角与上一行右上角无相邻的1

(  i & (j>>1) )==0 说明当前行右下角与上一行左上角无相邻的1

一行内不相邻的情况可以提前预处理出来，至于行与行之前的相邻情况在状态转移时特殊判断一下就行。

细节我都标在代码里了，下面上代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<queue>
using namespace std;
typedef long long ll;
const int N=103;
int n,k;
ll cnt;//cnt记录一行有多少种合法状态
ll king[10003];//king[i]记录第i中合法状态的国王数目
ll st[10003];//记录第i中合法状态 
ll dp[10][1<<12][101];//dp[i][j][k]表示第i行状态为j，且此时已摆放国王的数目为k
void init()
{
	for(int i=0;i<1<<n;i++)
		if((i&(i>>1))==0)//没有相邻的国王
		{
			st[++cnt]=i;
			int t=i;
			while(t)//记录i状态中国王数目 
			{
				king[cnt]+=t%2;
				t>>=1;
			}
		}
}
int main()
{
	cin>>n>>k;
	init();
	dp[0][0][0]=1;
	for(int i=1;i<=n;i++)
	for(int j=1;j<=cnt;j++)//枚举上一行状态 
	for(int l=1;l<=cnt;l++)//枚举当前行状态 
	{
		if(st[j]&st[l]) continue;//上下有相邻国王 
		if((st[j]>>1)&st[l]) continue;//左上右下有相邻国王 
		if(st[j]&(st[l]>>1)) continue;//右上左下有相邻国王 
		for(int s=0;s<=k;s++)//枚举上一行及之前所有行一共使用的国王数
		{
			if(s+king[l]>k) continue;
			dp[i][st[l]][s+king[l]]+=dp[i-1][st[j]][s];
		} 
	} 
	long long ans=0;	
	for(int i=1;i<=cnt;i++)
		ans+=dp[n][st[i]][k];
	printf("%lld",ans);
	return 0;
}

对于有些对复杂度要求高的题目就需要在状态转移前把合法状态先预处理出来，放到一个数组里面，不然容易超时，对于有些条件我们可以一开始就预处理掉的最好先预处理出来，这样在状态转移时不容易出错。最后要提醒大家的就是关于运算顺序的问题了，强烈建议大家在做状压dp时多加几个括号，不然如果出现了运算顺序的错误，就得不偿失了！，我就经常遇到这样错误，debug到怀疑人生！