[SCOI2005] 互不侵犯

时间:2023-03-08 22:12:33
[SCOI2005] 互不侵犯

传送门:>Here<

给出一个n*n的棋盘($n \leq 9$),放$k$个骑士,每个骑士可以攻击相邻的八个方向。问所有骑士互不侵犯的摆放方案数。

解题思路

决策问题可以通过搜索解决,而DP就是记忆化搜索。而在这里,我们直接考虑整排的决策比较方便。

在搜索时我们需要利用到哪些信息来完成决策?显然能影响到当前决策的有上一排的各个骑士位置,还能用几个骑士。而上一排的各个骑士位置是一个布尔数组,转化为DP的话这就成为了DP的一个状态。数据范围小的时候,我们是可以直接将布尔数组转为二进制作为状态的。我们称这种DP方法为状态压缩DP。

分析DP的时间复杂度,一般是状态数量乘上转移的复杂度。这里状态数是$O(2^nnk)$,而转移时枚举上一行状态$O(2^n)$,故总复杂度为$O(2^{2n}n^3)$。

这样的复杂度是过不了的。而事实上,一行内的合法状态数远不足$2^n$,所以我们可以预处理出每一行的合法状态数,这样就能过了。

$Code$

/*By QiXingzhi*/
#include <cstdio>
#define N (4010)
#define r read()
#define INF (0x3f3f3f3f)
#define Max(a,b) (((a)>(b)) ? (a) : (b))
#define Min(a,b) (((a)<(b)) ? (a) : (b))
typedef long long ll;
#define int ll
using namespace std;
inline int read(){
int x = ; int w = ; register int c = getchar();
while(c ^ '-' && (c < '' || c > '')) c = getchar();
if(c == '-') w = -, c = getchar();
while(c >= '' && c <= '') x = (x << ) +(x << ) + c - '', c = getchar();
return x * w;
}
int n,K,tot,ans;
int sta[N],num[N],f[][N][];
void Dfs(int x, int cur, int sum){
if(x >= n){
++tot;
sta[tot] = cur;
num[tot] = sum;
f[][tot][sum] = ;
return;
}
Dfs(x+,cur,sum);
Dfs(x+,cur+(<<x),sum+);
}
#undef int
int main(){
#define int ll
n=r,K=r;
Dfs(,,);
for(int i = ; i <= n; ++i){
for(int j = ; j <= tot; ++j){
for(int k = ; k <= tot; ++k){
if(sta[j] & sta[k]) continue;
if(sta[j] & (sta[k] << )) continue;
if(sta[j] & (sta[k] >> )) continue;
for(int s = num[j]; s <= K; ++s){
f[i][j][s] += f[i-][k][s-num[j]];
}
}
}
}
for(int i = ; i <= tot; ++i) ans += f[n][i][K];
printf("%lld",ans);
return ;
}