题目描述:
司令部的将军们打算在N*M的网格地图上部署他们的炮兵部队。一个N*M的地图由N行M列组成,地图的每一格可能是山地(用”H” 表示),也可能是平原(用”P”表示),如下图。
在每一格平原地形上最多可以布置一支炮兵部队(山地上不能够部署炮兵部队);一支炮兵部队在地图上的攻击范围如图中黑色区域所示:
如果在地图中的灰色所标识的平原上部署一支炮兵部队,则图中的黑色的网格表示它能够攻击到的区域:沿横向左右各两格,沿纵向上下各两格。
图上其它白色网格均攻击不到。从图上可见炮兵的攻击范围不受地形的影响。
现在,将军们规划如何部署炮兵部队,在防止误伤的前提下(保证任何两支炮兵部队之间不能互相攻击,即任何一支炮兵部队都不在其他支炮兵部队的攻击范围内),在整个地图区域内最多能够摆放多少我军的炮兵部队。
输入格式
第一行包含两个由空格分割开的正整数,分别表示N和M;
接下来的N行,每一行含有连续的M个字符(‘P’或者’H’),中间没有空格。按顺序表示地图中每一行的数据。
输出格式
仅一行,包含一个整数K,表示最多能摆放的炮兵部队的数量。
数据范围
N≤100,M≤10
输入样例:
5 4
PHPP
PPHH
PPPP
PHPP
PHHP
输出样例:
6
分析:
本题首先要注意的是读取地图上的字母最好要用cin读取而不能用scanf("%c"),这是因为scanf会连同每行末尾的回车符一起读进来,就算在各行末尾加上一个getchar(),还是可能多读取第一行数字后面的回车符,采用cin读取单个字符就不需要考虑这个问题了。另外,测试发现,尽管scanf读取字符串或者整型等数据要快于cin,但是可能一个个读取字符速度比cin要慢,容易出错速度又慢,因此,以后在读取单个字符时最好还是使用cin。
下面分析题意,标记有H的地方不能布置炮兵部队,这点快于参考上一题的做法,将各行不能放置的点存入一个二进制数,枚举状态时与该二进制数与一下,结果非0则不合法。然后是炮兵的攻击范围决定了同一行里放了1的相邻两个格子都不能在放1,即某一行放置炮兵的状态为s时,s & (s >> 1) 以及s & (s >> 2)都要为0。另外,枚举第i行状态时要受到第i-1行以及第i-2行状态的影响,设第i行状态为s,第i-1行状态为t,第i-2行状态为r,则这三行同一列最多只能有一个1,s & t,s & r,r & t都应该是0,注意不能简单的把s,t,r与起来判断是否为0,因为只要其中一个为0,与的结果都会是0。
由于每行的状态受前两行的影响,所以状态表示f[i][j][k]表示布置完前i行且第i行的状态为j,第i-1行的状态为k的最多能摆放炮兵的数量。状态转移方程为f[i][j][k] = max(f[i][j][k],f[i-1][k][l] + cnt[j]),其中l为第i - 2行的状态,cnt[j]为第i行的状态j中1的数量。具体实现时状态转移有较大的出入,一方面是因为空间限制64MB,而2^20约等于10^6,如果数组有10^8个整型元素,则大概是4 * 100 * 10^6 = 400 * 2^20B = 400MB,本题的f数组有100 * 2^10 * 2^10 * 4= 400MB超过了64MB,可以使用滚动数组,第一维仅用两行存储状态,这样状态数是2 * 4 MB = 8MB。第i行的状态可以放在u = i & 1行,第i-1行的状态就放在v = !u行即可。另一方面是我们需要预处理合法的 状态,最后需要枚举的状态存储在一个向量里,向量的下标与状态一一对应,所以f[i][j][k]的状态j和k可以用他们在向量里的下标表示。虽然需要注意的地方不少,但是实现起来代码还是相当简单的。
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
const int N = 105,M = 1 << 11;
int n,m,f[2][M][M],g[N],cnt[M];
vector<int> a;
bool judge(int st){
int s = st & (st >> 1),t = st & (st >> 2);
return (s | t) == 0;
}
int count(int st){
int res = 0;
for(int i = 0;i < m;i++) res += 1 & st >> i;
return res;
}
int main(){
cin>>n>>m;
char c;
for(int i = 1;i <= n;i++){
for(int j = m - 1;j >= 0;j--){
cin>>c;
if(c == 'H') g[i] += 1 << j;
}
}
for(int i = 0;i < 1 << m;i++){
if(judge(i)) a.push_back(i),cnt[i] = count(i);
}
for(int i = 1;i <= n;i++){
int u = i & 1,v = !u;
for(int j = 0;j < a.size();j++){
if(g[i] & a[j]) continue;
for(int k = 0;k < a.size();k++){
if(g[i - 1] & a[k]) continue;
for(int l = 0;l < a.size();l++){
if(i > 2 && g[i-2] & a[l]) continue;
int s = a[j],t = a[k],r = a[l];
if((s & t | s & r | t & r) == 0) f[u][j][k] = max(f[u][j][k],f[v][k][l] + cnt[s]);
}
}
}
}
int res = 0,u = n & 1;
for(int i = 0;i < a.size();i++){
for(int j = 0;j < a.size();j++){
res = max(res,f[u][i][j]);
}
}
cout<<res<<endl;
return 0;
}