很早之前就学习过状压了吧,只不过当时完全没怎么做过题,甚至连这最基础的几道都没做过。今天翻出来做了,做了这道感觉受益匪浅。
传送门
题目描述
司令部的将军们打算在NM的网格地图上部署他们的炮兵部队。一个NM的地图由N行M列组成,地图的每一格可能是山地(用“H” 表示),也可能是平原(用“P”表示),如下图。在每一格平原地形上最多可以布置一支炮兵部队(山地上不能够部署炮兵部队);一支炮兵部队在地图上的攻击范围如图中黑色区域所示:
如果在地图中的灰色所标识的平原上部署一支炮兵部队,则图中的黑色的网格表示它能够攻击到的区域:沿横向左右各两格,沿纵向上下各两格。图上其它白色网格均攻击不到。从图上可见炮兵的攻击范围不受地形的影响。 现在,将军们规划如何部署炮兵部队,在防止误伤的前提下(保证任何两支炮兵部队之间不能互相攻击,即任何一支炮兵部队都不在其他支炮兵部队的攻击范围内),在整个地图区域内最多能够摆放多少我军的炮兵部队。
输入格式
第一行包含两个由空格分割开的正整数,分别表示N和M;
接下来的N行,每一行含有连续的M个字符(‘P’或者‘H’),中间没有空格。按顺序表示地图中每一行的数据。N≤100;M≤10。
输出格式
仅一行,包含一个整数K,表示最多能摆放的炮兵部队的数量。
输入输出样例
输入 #1
5 4
PHPP
PPHH
PPPP
PHPP
PHHP
输出 #1
6
看数据发现n≤100,而m≤10,看m那么小便考虑状压dp,但n又过大。然后发现一个炮兵部队只对上下2行起作用,便是说dp时只用考虑上2层的状态,便考虑设f[i][a][b][c],a是本行状态,b是上行,c是上2行,转移便是从f[i-1][b][c][d],这便是把i-2行放入f中。但不仅会远远超出空间范围,而且在后面的状态转移方程中我们可以发现不需要考虑第i-2行,在进行第i-1的转移时已经考虑过了。
所以我们设f[i][j][k] 表示最大个数,f[i][j][k] = max(f[i][j][k] ,f[i - 1][k][l] + sum[a[i][j]])
j枚举当前行状态,k枚举i - 1行状态,l枚举i - 2行状态
sum记下不同状态的炮兵数。
然后注意行内炮兵之间不冲突,行间也是。
具体做法:
1.预处理:
每行地形s[]
每种状态多少个1 sum[]
单论一行中不冲突的状态数num[],与状态b[][]
第一行和第二行的f单独计算。
2.枚举i行,i行状态,i-1行状态,i-2行状态,判断是否冲突后更新f
3.从第n行的f得出ans。
tips:因为即使f数组我们少了一维,但也容易MLE,发现在行那一维每次只跟上一行有关,可以通过滚动数组减少空间复杂度。
MyCode(1.25KB,139ms):
#include<bits/stdc++.h>
using namespace std;
int n,m,f[1024][1024][2],t,num[101],b[101][1024],s[101],ans,sum[1024];
char x;
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
cin>>x;
s[i]=(s[i]<<1)+(x=='H');
}
for(int i=1;i<=(1<<m)-1;i++)
{
for(int j=0;j<m;j++)
if((1<<j)&i) sum[i]++;
}
for(int i=1;i<=n;i++)
{
for(int j=0;j<=(1<<m)-1;j++)
{
if(!(j&s[i])&&!(j&(j<<1))&&!(j&(j<<2)))
num[i]++,b[i][num[i]]=j;
}
}
for(int i=1;i<=num[1];i++) f[0][b[1][i]][t]=sum[b[1][i]];
t^=1;
for(int i=1;i<=num[1];i++)
{
for(int j=1;j<=num[2];j++)
{
if(!(b[1][i]&b[2][j])) f[b[1][i]][b[2][j]][t]=max(f[b[1][i]][b[2][j]][t],f[0][b[1][i]][t^1]+sum[b[2][j]]);
}
}
for(int i=3;i<=n;i++)
{
t^=1;
for(int j=1;j<=num[i];j++)
{
for(int k=1;k<=num[i-1];k++)
{
for(int l=1;l<=num[i-2];l++)
{
if(!(b[i][j]&b[i-2][l])&&!(b[i-1][k]&b[i-2][l]))
f[b[i-1][k]][b[i][j]][t]=max(f[b[i-1][k]][b[i][j]][t],f[b[i-2][l]][b[i-1][k]][t^1]+sum[b[i][j]]);
}
}
}
}
for(int i=1;i<=num[n];i++)
{
for(int j=1;j<=num[n-1];j++)
{
if(!(b[n][i]&b[n-1][j]))
ans=max(ans,f[b[n-1][j]][b[n][i]][t]);
}
}
cout<<ans;
return 0;
}