[SCOI2005]互不侵犯

题目传送门

题面

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

解题方法

状压dp入门题目，使用二进制压缩每个位置上的国王摆放情况，然后转移$dp[i][j][k]$，表示在第$i$行放了$k$个国王，其中第$i$行状态是$j$的种类数。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int n,rs;
int tot;
long long dp[10][2000][100];
int zt[2001],s[2001];
void dfs(int z,int sum,int w)
{
	if(w>=n)
	{
		tot++;
		zt[tot]=z;
		s[tot]=sum;
		return;
	}
	dfs(z,sum,w+1);
	dfs(z+(1<<w),sum+1,w+2);
}
int main()
{	
	scanf("%d%d",&n,&rs);
	dfs(0,0,0);
	for(int i=1;i<=tot;i++)
	  dp[1][i][s[i]]=1;
	for(int i=2;i<=n;i++)
	  for(int j=1;j<=tot;j++)
	    for(int k=1;k<=tot;k++)
	    {
	    	if(zt[j]&zt[k])continue;
	    	if((zt[j]<<1)&zt[k])continue;
	    	if(zt[j]&(zt[k]<<1))continue;
	    	for(int t=rs;t>=s[j];t--)
	    	  dp[i][j][t]+=dp[i-1][k][t-s[j]];
		}
	long long ans=0;
	for(int i=1;i<=tot;i++)
	  ans+=dp[n][i][rs];
	cout<<ans<<'\n';
	return 0;
}