链接:登录—专业IT笔试面试备考平台_牛客网
来源:牛客网
题目描述
司令部的将军们打算在N*M的网格地图上部署他们的炮兵部队。一个N*M的地图由N行M列组成,地图的每一格可能是山地(用"H" 表示),也可能是平原(用"P"表示),如下图。在每一格平原地形上最多可以布置一支炮兵部队(山地上不能够部署炮兵部队);一支炮兵部队在地图上的攻击范围如图中黑色区域所示:
如果在地图中的灰色所标识的平原上部署一支炮兵部队,则图中的黑色的网格表示它能够攻击到的区域:沿横向左右各两格,沿纵向上下各两格。图上其它白色网格均攻击不到。从图上可见炮兵的攻击范围不受地形的影响。
现在,将军们规划如何部署炮兵部队,在防止误伤的前提下(保证任何两支炮兵部队之间不能互相攻击,即任何一支炮兵部队都不在其他支炮兵部队的攻击范围内),在整个地图区域内最多能够摆放多少我军的炮兵部队。
输入描述:
第一行包含两个由空格分割开的正整数,分别表示N和M; 接下来的N行,每一行含有连续的M个字符('P'或者'H'),中间没有空格。按顺序表示地图中每一行的数据。N≤100;M≤10。
输出描述:
一行,包含一个整数K,表示最多能摆放的炮兵部队的数量
示例1
输入
5 4 PHPP PPHH PPPP PHPP PHHP
输出
6
分析:
首先,我们可以将收山地图抽象为01串的状态来表示(如0表示平原,1表示山地),然后,我们可以按行进行转移,当前行的状态与前一行和前前一行的状态有关,则可以定义dp式子为f【i】【j】【k】表示第i行当前行状态为j最多能放置的数量,前一行的状态为k,转移方程为:
f【i】【j】【k】=max(f【i】【j】【k】,f【i-1】【k】【p】+num【j】);
合法转移条件有:
j&s[i]==0(不能再H山地上放置)
j&(j<<1)==0 ,j&(j<<2)==0;(每行相邻两格之内不能同时放置)
j&k==0;(当前行和上一行不能同时放置)
j&p==0;(当前行和上上一行不能同时放置)
num【j】表示状态为j时放置的数量(1的数量)
对于num数组和每行的合法状态(最多60个状态)可以提前打表求出,枚举状态可能会超时。
参考代码:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
int s[105];
ll f[105][70][70];
int num[70];
int cal(int x)
{
int cnt=0;
while(x)
{
if(x&1)cnt++;
x>>=1;
}
return cnt;
}
int main ()
{
int n,m;cin>>n>>m;
for(int i=1;i<=n;i++)
for(int j=0;j<m;j++)
{
char c;
cin>>c;
if(c=='H')s[i]^=(1<<j);
}
vector<int>q;
for(int i=0;i<(1<<m);i++)if(((i<<1)&i)==0&&((i<<2)&i)==0)q.pb(i);
for(int i=0;i<q.size();i++)num[i]=cal(q[i]);
ll ans=0;
for(int i=1;i<=n;i++)
{
for(int j=0;j<q.size();j++)
{
if(q[j]&s[i])continue;
for(int k=0;k<q.size();k++)
{
for(int p=0;p<q.size();p++)
{
if(q[j]&q[k])continue;
if(q[j]&q[p])continue;
f[i][j][k]=max(f[i][j][k],f[i-1][k][p]+num[j]);
ans=max(ans,f[i][j][k]);
}
}
}
}
cout<<ans;
return 0;
}