Description
司令部的将军们打算在N*M的网格地图上部署他们的炮兵部队。一个N*M的地图由N行M列组成,地图的每一格可能是山地(用”H” 表示),也可能是平原(用”P”表示),如下图。在每一格平原地形上最多可以布置一支炮兵部队(山地上不能够部署炮兵部队);一支炮兵部队在地图上的攻击范围如图中黑色区域所示:
如果在地图中的灰色所标识的平原上部署一支炮兵部队,则图中的黑色的网格表示它能够攻击到的区域:沿横向左右各两格,沿纵向上下各两格。图上其它白色网格均攻击不到。从图上可见炮兵的攻击范围不受地形的影响。现在,将军们规划如何部署炮兵部队,在防止误伤的前提下(保证任何两支炮兵部队之间不能互相攻击,即任何一支炮兵部队都不在其他支炮兵部队的攻击范围内),在整个地图区域内最多能够摆放多少我军的炮兵部队。
Input
第一行包含两个由空格分割开的正整数,分别表示N和M; 接下来的N行,每一行含有连续的M个字符(‘P’或者’H’),中间没有空格。按顺序表示地图中每一行的数据。N <= 100;M <= 10。
Output
仅一行,包含一个整数K,表示最多能摆放的炮兵部队的数量。
Sample Input
5 4
PHPP
PPHH
PPPP
PHPP
PHHP
Sample Output
6
Solution
状态压缩DP,首先显然是通过一个二进制数来表示一行的状态,然后通过前两行来推出下一行的状态,但考虑到数据范围,如果每次都暴力枚举一行所有状态会超时,所以在状态数上需要优化,因为每行最多10格,相邻两支炮兵部队最小间距为2,所以合法状态不会超过60种,这样的话我们可以预处理每行都为空地时的所有合法状态,每行枚举状态时就极大的降低了时间复杂度。
下面看具体操作,用dp[h][i][j]表示第h行状态为i,第h-1行状态为j时的炮兵总数,首先算出前两行的所有合法状态,之后由递推式dp[h][i][j]=max{dp[h][i][j],dp[h-1][j][k]+num[i]}(k为第h-2行的状态,num[i]表示第h行状态为i时可以放置的炮兵数)就可以递推出所有行的合法状态,这样最多能放置的数量即为max{dp[n-1][i][j]}(0<=i< p,0<=j< p,p为所有合法状态)
Code
#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
#define maxn 111
int map[maxn];
int dp[maxn][maxn][maxn];
int s[maxn],num[maxn];
int n,m,p;
bool check(int x)//判断状态x是否会造成行冲突
{
if(x&(x>>1))return false;
if(x&(x>>2))return false;
return true;
}
int count(int x)//统计x状态需要放置多少炮兵
{
int i=1,ans=0;
while(i<=x)
{
if(x&i)//该位置放置炮兵
ans++;
i<<=1;
}
return ans;
}
void init()
{
//初始化
p=0;
memset(s,0,sizeof(s));//s数组存所有合法状态
memset(num,0,sizeof(num));//num数组存所有合法状态的炮兵数
for(int i=0;i<(1<<m);i++)//枚举所有可能状态
{
if(check(i))//i状态合法
{
s[p]=i;//记录
num[p++]=count(i);//统计i状态需要放置炮兵数
}
}
}
int main()
{
while(scanf("%d%d",&n,&m)!=EOF)
{
if(!n&&!m)//有一边为0结果显然为0
{
printf("0\n");
continue;
}
//初始化
memset(dp,0,sizeof(dp));
memset(map,0,sizeof(map));
for(int i=0;i<n;i++)//输入
for(int j=0;j<m;j++)
{
char ch;
cin>>ch;
if(ch=='H')
map[i]=map[i]|(1<<(m-1-j));//只有P才能放炮兵
}
init();//预处理
for(int i=0;i<p;i++)//计算第一行的状态
{
if(!(map[0]&s[i]))
dp[0][i][0]=num[i];
}
for(int i=0;i<p;i++)//计算第二行的状态
{
if(!(map[1]&s[i]))
{
for(int j=0;j<p;j++)
{
if(!(s[i]&s[j]))
dp[1][i][j]=max(dp[1][i][j],dp[0][j][0]+num[i]);
}
}
}
for(int h=2;h<n;h++)//计算第2~n-1行的状态
{
for(int i=0;i<p;i++)//枚举第h行状态
{
if(!(map[h]&s[i]))//由地形排除不合法状态
{
for(int j=0;j<p;j++)//枚举第h-1行状态
{
if(!(map[h-1]&s[j]))//由地形排除不合法状态
{
if(!(s[i]&s[j]))//第h行与第h-1行不能冲突
{
for(int k=0;k<p;k++)//枚举第h-2行状态
{
if(!(map[h-2]&s[k]))//由地形排除不合法状态
{
if(!(s[i]&s[k]))//第h行不能与第h-2行冲突
{
if(!(s[j]&s[k]))//第h-1行不能与第h-2行冲突
{
dp[h][i][j]=max(dp[h][i][j],dp[h-1][j][k]+num[i]);
}
}
}
}
}
}
}
}
}
}
int ans=0;
for(int i=0;i<p;i++)
for(int j=0;j<p;j++)
if(ans<dp[n-1][i][j])
ans=dp[n-1][i][j];
printf("%d\n",ans);
}
return 0;
}