BZOJ1087: [SCOI2005]互不侵犯King 题解【DP】【状压】

时间:2022-12-17 17:57:51

【题目描述】
在N×N的棋盘里面放K个国王,使他们互不攻击,共有多少种摆放方案。国王能攻击到它上下左右,以及左上
左下右上右下八个方向上附近的各一个格子,共8个格子。
【题解】
这道题一看就给人一种八皇后问题的感觉,然后我就想到了搜索,然后就TLE了一次。所以在人们太空军政委同志(这是他的博客)的教导下,我意识到这是状压DP。
这道题的思路是这样的:我们先把第一行可行的方案枚举出来,用二进制表示,然后套用我们的转移方程。
代码如下:

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;
int n,m,k,stay[101],cnt[101];
long long ans,dp[10][101][101];
bool map[101][101];
void dfs(int p,int put,int num)
{
stay[++m]=num;
cnt[m]=p;
if(p>=k||p>=(n+1)/2)return;
for(int i=put+2;i<=n;i++)
dfs(p+1,i,num+(1<<(i-1)));
}
int main()
{
scanf("%d%d",&n,&k);
dfs(0,-1,0);
for(int i=1;i<=m;i++)
for(int j=1;j<=m;j++)
map[i][j]=map[j][i]=((stay[i]&stay[j])||((stay[i]<<1)&stay[j])||((stay[i]>>1)&stay[j]))?0:1;
for(int i=1;i<=m;i++)
dp[1][cnt[i]][i]=1ll;
for(int i=2;i<=n;i++)
for(int j=0;j<=k;j++)
for(int now=1;now<=m;now++)
{
if(cnt[now]>j)continue;
for (int l=1;l<=m;l++)
if (map[now][l]&&cnt[l]+cnt[now]<=j) dp[i][j][now]+=dp[i-1][j-cnt[now]][l];
}
for(int i=1;i<=m;i++)ans+=dp[n][k][i];
printf("%lld",ans);
return 0;
}