状态压缩动态规划和状态机动态规划(未更新)大体上来说都是通过另类的方式来储存动态规划中的状态。状压DP其实可以分为“棋盘式”和“集合类”,这篇是“棋盘式”或者叫“基于联通性”的状压DP的一道经典例题——炮兵阵地的博客。
炮兵阵地这道非常经典的状态压缩DP题目是值得学习的,可以从中学习到“棋盘式”状压DP的一些基本操作和标准化思想。
题目:
司令部的将军们打算在
N
∗
M
N*M
N∗M 的网格地图上部署他们的炮兵部队。一个
N
∗
M
N*M
N∗M 的地图由
N
N
N行
M
M
M 列组成,地图的每一格可能是山地(用”H” 表示),也可能是平原(用”P”表示),如下图。
在每一格平原地形上最多可以布置一支炮兵部队(山地上不能够部署炮兵部队);一支炮兵部队在地图上的攻击范围如图中黑色区域所示:
如果在地图中的灰色所标识的平原上部署一支炮兵部队,则图中的黑色的网格表示它能够攻击到的区域:沿横向左右各两格,沿纵向上下各两格。
图上其它白色网格均攻击不到。
从图上可见炮兵的攻击范围不受地形的影响。
现在,将军们规划如何部署炮兵部队,在防止误伤的前提下(保证任何两支炮兵部队之间不能互相攻击,即任何一支炮兵部队都不在其他支炮兵部队的攻击范围内),在整个地图区域内最多能够摆放多少我军的炮兵部队。
动规做法:
1、状态表示:
f
[
i
]
[
j
]
[
k
]
f[i][j][k]
f[i][j][k] 表示在更新第
i
i
i 行时第
i
−
1
i-1
i−1 和
i
−
2
i-2
i−2 行的状态分别为
j
、
k
j、k
j、k 的炮兵数量
这里
j
j
j 和
k
k
k 是一个数字,在二进制下表示下1代表这里可以放,而0是不能放(因为受上面的炮影响)。
2、状态转移:从第一行向下递推
f
[
i
]
[
j
]
[
k
]
=
m
a
x
{
f
[
i
−
1
]
[
k
]
[
p
]
+
c
o
u
n
t
(
j
)
}
f[i][j][k]=max\{f[i-1][k][p]+count(j)\}
f[i][j][k]=max{f[i−1][k][p]+count(j)}
j
j
j 表示第
i
−
1
i-1
i−1 行的状态,
k
k
k 表示第
i
−
2
i-2
i−2 行的状态,
p
p
p 表示第
i
−
3
i-3
i−3 行的状态,
c
o
u
n
t
(
j
)
count(j)
count(j)表示状态
j
j
j 中又多少个1。
当然,这里的状态转移重要的是“合法”二字,对于“合法”要从两个方面入手:
1)要从合法的状态转移到合法的状态,对于单行状态合法性的判断就是要求两个1之间至少隔两个0:
bool check(int x){
for(int i=0;i<m;i++)
if( (x>>i & 1) && ((x>>i+1 & 1) || (x>>i+2 & 1))) //1与1之间至少两个空格
return false;
return true;
}
2)就是转移规则要合法,这一点有两点限制:
a)地形要合法(这里为了方便位运算,地形的存储数字可以放是0,不可以是1)
b)该排的炮兵不能在前两排的范围内
if(ori[i]&a | ori[i-1]&b) continue;
if((a&c) | (a&b) | (b&c)) continue;
最后在第
i
i
i 行的所有合法状态中找最大值即可。
注意,由于数据范围,这里
f
f
f 数组需要用滚动数组储存。
完整代码:
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int N=110,M=1<<10;
int f[2][M][M];
vector<int> state;
int cnt[M],ori[N];
int n,m;
bool check(int x){
for(int i=0;i<m;i++)
if( (x>>i & 1) && ((x>>i+1 & 1) || (x>>i+2 & 1))) //1与1之间至少两个空格
return false;
return true;
}
int count(int x){
int res=0;
for(int i=0;i<m;i++)
if(x>>i & 1) res++;
return res;
}
int main(){
cin>>n>>m;
for(int i=1;i<=n;i++){
for(int j=0;j<m;j++){
char c;
cin>>c;
ori[i]+=(c == 'H') << j; //为了后面检验可以用快速的位运算,这里写不能放的地方是1,可以放的是0才行,要不后面再判断会超时
}
//cout<<ori[i]<<endl;
}
for(int i=0;i<1<<m;i++)
if(check(i)){
state.push_back(i);
cnt[i]=count(i);
//cout<<cnt[i]<<endl;
}
//ori[0]=(1<<m)-1;
for(int i=1;i<=n;i++){
for(int j=0;j<state.size();j++){ //i
for(int k=0;k<state.size();k++){ //i-1
for(int u=0;u<state.size();u++){ //i-2
int a=state[j],b=state[k],c=state[u];
if(ori[i]&a | ori[i-1]&b) continue;
if((a&c) | (a&b) | (b&c)) continue;
f[i&1][a][b]=max(f[i&1][a][b],f[i-1&1][b][c]+cnt[a]);
}
}
}
}
int res=0;
for(int i=0;i<state.size();i++)
for(int j=0;j<state.size();j++)
res=max(res,f[n&1][state[i]][state[j]]);
cout<<res<<endl;
return 0;
}