题目:
司令部的将军们打算在NM的网格地图上部署他们的炮兵部队。一个NM的地图由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
代码如下:
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
using namespace std;
int dp[105][200][200];//dp[i][j][k]代表第i行状态为j时,i - 1行状态为k时可放炮台的个数
int cur[200];//第i行可能的状态
int sum[200];//这种状态放的炮台数目
int top,state[200];//保存第一行的所有可行方案
int n,m,maxn = -1;
bool ok(int i) //防止出现设置的炮台出现在已有炮台的射程范围内
{
if(i & (i << 1)) return false;
if(i & (i << 2)) return false;
return true;
}
int get_num(int i) //计算出这总状态中炮台数量
{
int ans = 0;
while(i > 0){
if(i & 1) ans++;
i >>= 1;
}
return ans;
}
void init()
{
top = 0;
memset(state,0,sizeof(state));
int num = (1 << m);
for(int i = 0;i < num;i++){
if(ok(i)){
state[top] = i;
sum[top++] = get_num(i); //记录每种可行状态的炮台数目
}
}
}
int main()
{
while(~scanf("%d%d",&n,&m)){
memset(dp,-1,sizeof(dp));
for(int i = 0;i < n;i++){
cur[i] = 0;
for(int j = 0;j < m;j++){
char temp;
cin >> temp;
if(temp == 'H') cur[i] += (1 << (m - j - 1));//H代表是高地不能放炮台,所以设置为1
}
}
init();
for(int i = 0;i < top;i++) if(!(state[i] & cur[0])) dp[0][i][0] = sum[i];//第一行每种状态的炮台数
for(int r = 1;r < n;r++){
for(int i = 0;i < top;i++){
if(cur[r] & state[i]) continue;//第i种状况和第r列不匹配
for(int j = 0;j < top;j++){
if(state[i] & state[j]) continue;
for(int k = 0;k < top;k++){
if(state[j] & state[k]) continue;//这一行炮台在在射程范围内
if(state[i] & state[k]) continue;
if(dp[r - 1][j][k] == -1) continue;
dp[r][i][j] = max(dp[r][i][j],dp[r - 1][j][k] + sum[i]); //每次第r行的状态i都选取最大值
}
}
}
}
for(int i = 0;i < top;i++)
for(int j = 0;j < top;j++)
maxn = max(maxn,dp[n - 1][i][j]);
cout << maxn << endl;
}
return 0;
}
这道题和poj3254放牛这道题差不多,都是用状压dp去处理。唯一不同是poj3254是求总共放牛的种类,而这道题求的是放炮塔最多数量。求最大数量只需从放到最后一行的所有状态中选一个最大的就可以了,具体过程,注释有。