题目链接
[NOI2001]炮兵阵地
分析
此题是一道经典的状压
d
p
dp
dp,需要注意的是要考虑当前行的前两行的状态,因此需要开三维数组,令
d
p
[
i
]
[
j
]
[
k
]
dp[i][j][k]
dp[i][j][k]表示枚举到第
i
i
i行时第
i
−
1
i-1
i−1行和
i
−
2
i-2
i−2行的二进制状态分别为
j
j
j、
k
k
k的最大摆放数,则有递推方程
d
p
[
i
]
[
j
]
[
k
]
=
max
(
d
p
[
i
]
[
j
]
[
k
]
,
d
p
[
i
−
1
]
[
k
]
[
l
]
+
c
n
t
)
dp[i][j][k] = \max (dp[i][j][k],dp[i - 1][k][l] + cnt)
dp[i][j][k]=max(dp[i][j][k],dp[i−1][k][l]+cnt),其中
c
n
t
cnt
cnt为当前行放的数量(满足条件的话)。此题思路不复杂,但是处理初始化条件时需要格外注意,另外考虑到
d
p
dp
dp数组每次更新只与
d
p
[
i
−
1
]
dp[i-1]
dp[i−1]有关,所以可以用滚动数组优化。
代码
#include<bits/stdc++.h>
#define ll long long
#define FULL(x,y) memset(x,y,sizeof(x))
#define pb push_back
using namespace std;
int n,m;
char c[105][11];
int dp[2][1025][1025],f[1025];
int main() {
cin>>n>>m;
for(int i=1;i<=n;i++) {
for(int j=1;j<=m;j++) {
cin>>c[i][j];
}
}
//初始化
int all=1<<m,ans=0;
for(int i=0;i<all;i++) {
for(int j=0;j<all;j++) {
dp[0][i][j]=0;
}
}
for(int i=0;i<all;i++) {
if (i&(i<<1) || i&(i<<2) || i&(i>>1) || i&(i>>2)) f[i]=1;
}
for(int i=1;i<=n;i++) {//枚举行数
for(int j=all-1;j>=0;j--) {
if (f[j]) continue;
int fl=0,cnt=0;
for(int k=0;k<m;k++) {
if (j&(1<<k)) {
if (c[i][k+1]=='H') {
fl=1;
break;
}
else cnt++;
}
}
if (fl) continue;
for(int k=all-1;k>=0;k--) {
if (i==1) {
dp[i&1][j][k]=cnt;
continue;
}
if (f[k] || j&k) continue;
for(int l=all-1;l>=0;l--) {
if ((i==2 && dp[1-(i&1)][k][l]!=-1) || (i>2 && dp[1-(i&1)][k][l]!=-1 && (j&l)==0)) {
dp[i&1][j][k]=max(dp[i&1][j][k],dp[1-(i&1)][k][l]+cnt);
}
}
ans=max(ans,dp[i&1][j][k]);
}
}
}
cout<<ans;
return 0;
}