洛谷P1896 [SCOI2005]互不侵犯King

题目描述

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

输入输出格式

输入格式：

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

输出格式：

所得的方案数

输入输出样例

输入样例#1： 

3 2

输出样例#1： 

16

状态压缩DP

设 dp[i][j][k] 表示第i行的状态为j，前i行一共用了k个国王

预先处理出每一个状态（a[x]）其中包含二进制下1的个数，及此状态下这一行放的国王个数（b[x]）

转移：$f[i][j][s]=sum(f[i-1][k][s-b[j]])$ 

判定局面合法：if(a[j]&a[k])||((a[j]<<1)&a[k])||(a[j]&(a[k]<<1)))continue;

附代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
int n,m,S=0,a[(1<<10)+1],b[(1<<10)+1];
long long ans=0,dp[10][(1<<10)+1][100];
inline int read(){
	int date=0,w=1;char c=0;
	while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();}
	while(c>='0'&&c<='9'){date=date*10+c-'0';c=getchar();}
	return date*w;
}
void dfs(int x,int s,int sum){
	if(x>=n){
		a[++S]=s;
		b[S]=sum;
		return;
	}
	dfs(x+1,s,sum);
	dfs(x+2,s+(1<<x),sum+1);
}
int main(){
	n=read();m=read();
	dfs(0,0,0);
	for(int s=1;s<=S;s++)dp[1][s][b[s]]=1;
	for(int i=2;i<=n;i++)
	for(int j=1;j<=S;j++)
	for(int k=1;k<=S;k++){
		if((a[j]&a[k])||((a[j]<<1)&a[k])||(a[j]&(a[k]<<1)))continue;
		for(int s=m;s>=b[j];s--)dp[i][j][s]+=dp[i-1][k][s-b[j]];
	}
	for(int s=1;s<=S;s++)ans+=dp[n][s][m];
	printf("%lld\n",ans);
	return 0;
}