[SCOI2005]互不侵犯 解题报告


题目描述

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

输入输出格式
输入格式:

只有一行,包含两个数N,K ( 1 <=N <=9, 0 <= K <= N * N)

输出格式:

所得的方案数

输入输出样例:
input

3 2

output

16

思路:

N的个数是十分小的 (即使这样爆搜也会T飞!!)
所以用一波 状压DP!

设f[i][j][k]代表前i行放了k个,排列方式是j,(j的二进制形式就是排列方式)。

可以 (必须) 先预处理出一行的可行排列情况:
1
2
3
4
5
6
7
8
9
10
11
void dfs(int w,int sum,int num)//分别为排列情况,放的点的个数,到底几个点了
{
if(num>=n)
{
sit[++cnt]=w;
gs[cnt]=sum;
return;
}
dfs(w,sum,num+1);//不防这个点
dfs(w+(1<<num),sum+1,num+2);//放这个点,那么下一个点不能放
}

那么根据题意,这个方案与另一个方案是否冲突 就显而易见啦
sit[j]&sit[k]==1 上下相邻
(sit[j]<<1)&sit[k]==1 斜着相邻
sit[j]&(sit[k]<<1) ==1 同上

DP+判断 code:

第一行:到第几行了
第二行:这一行的的排列情况
第三行:这一行的的排列情况可以从上一行的哪几个状态转移过来。

1
2
3
4
5
6
7
8
9
for(int i=2;i<=n;i++)
for(int j=1;j<=cnt;j++)
for(int k=1;k<=cnt;k++)
{
if(sit[j]&sit[k])continue;
if((sit[j]<<1)&sit[k]) continue;
if(sit[j]&(sit[k]<<1)) continue;
for(int s=K;s>=gs[j];s--) f[i][j][s]+=f[i-1][k][s-gs[j]];
}

完整代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
#include<bits/stdc++.h>
using namespace std;
#define maxn 2005

int cnt,n,K;
int sit[maxn],gs[maxn];
long long f[10][maxn][105];
long long ans;

void dfs(int w,int sum,int num)
{
if(num>=n)
{
sit[++cnt]=w;
gs[cnt]=sum;
return;
}
dfs(w,sum,num+1);
dfs(w+(1<<num),sum+1,num+2);
}

int main()
{
scanf("%d%d",&n,&K);
dfs(0,0,0);
for(int i=1;i<=cnt;i++)
f[1][i][gs[i]]=1;
for(int i=2;i<=n;i++)
for(int j=1;j<=cnt;j++)
for(int k=1;k<=cnt;k++)
{
if(sit[j]&sit[k])continue;
if((sit[j]<<1)&sit[k]) continue;
if(sit[j]&(sit[k]<<1)) continue;
for(int s=K;s>=gs[j];s--) f[i][j][s]+=f[i-1][k][s-gs[j]];
}
ans=0;
for(int i=1;i<=cnt;i++) ans+=f[n][i][K];
printf("%lld",ans);
return 0;
}

注意开long long。。。


Author: BY 水蓝
Reprint policy: All articles in this blog are used except for special statements CC BY 4.0 reprint polocy. If reproduced, please indicate source BY 水蓝 !
  TOC