司令部的将军们打算在N*M的网格地图上部署他们的炮兵部队。一个N*M的地图由N行M列组成,地图的每一格可能是山地(用"H" 表示),也可能是平原(用"P"表示),如下图。在每一格平原地形上最多可以布置一支炮兵部队(山地上不能够部署炮兵部队);一支炮兵部队在地图上的攻击范围如图中黑色区域所示:
如果在地图中的灰色所标识的平原上部署一支炮兵部队,则图中的黑色的网格表示它能够攻击到的区域:沿横向左右各两格,沿纵向上下各两格。图上其它白色网格均攻击不到。从图上可见炮兵的攻击范围不受地形的影响。
如果在地图中的灰色所标识的平原上部署一支炮兵部队,则图中的黑色的网格表示它能够攻击到的区域:沿横向左右各两格,沿纵向上下各两格。图上其它白色网格均攻击不到。从图上可见炮兵的攻击范围不受地形的影响。
现在,将军们规划如何部署炮兵部队,在防止误伤的前提下(保证任何两支炮兵部队之间不能互相攻击,即任何一支炮兵部队都不在其他支炮兵部队的攻击范围内),在整个地图区域内最多能够摆放多少我军的炮兵部队。
状压dp模版题的加强,因为本层会受上两层的影响,所以需要开三维dp,分别是层数,上一层状态,本层状态:
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<iostream>
using namespace std;
const int MOD=1000000007;
typedef long long ll;
int dp[105][65][65];
int mp[105]={0};//存图
int state[65];//在没有山头的情况下,满足的状态
int sum[65];//每个状态下炮兵的个数
int t;
bool judge(int x)
{
if(x&(x<<1)) return 0;
if(x&(x<<2)) return 0;
return 1;
}
int get_sum(int x)
{
int ans=0;
while(x)
{
if(x&1) ans++;
x>>=1;
}
return ans;
}
int get_state(int m)
{
int maxn=1<<m;
for(int i=0;i<maxn;i++)
{
if(judge(i))
{
state[t]=i;
sum[t++]=get_sum(i);
}
}
}
int main()
{
int n,m;
memset(dp,-1,sizeof(dp));
cin>>n>>m;
for(int i=0;i<n;i++)
{
for(int j=m-1;j>=0;j--)
{
char ch;
cin>>ch;
if(ch=='H') mp[i]|=(1<<j);
}
}
//for(int i=0;i<n;i++)
// printf("%d\n",mp[i]);
//printf("do!");
get_state(m);
//printf("do!");
for(int i=0;i<t;i++)
{
if((mp[0]&state[i])==0)
dp[0][0][i]=sum[i];
}
//printf("do!");
for(int r=1;r<n;r++)
{
for(int i=0;i<t;i++)
{
if(mp[r]&state[i]) continue;
for(int j=0;j<t;j++)
{
if(state[j]&state[i]) continue;
for(int k=0;k<t;k++)
{
if(state[k]&state[i]) continue;
if(dp[r-1][k][j]==-1) continue;
dp[r][j][i]=max(dp[r-1][k][j]+sum[i],dp[r][j][i]);
}
}
}
}
int ans = 0;
for(int i = 0; i < t; i ++)
for(int j = 0; j < t; j ++)
ans = max(ans, dp[n-1][i][j]);
printf("%d\n",ans);
return 0;
}