OJ题目 : 戳这里~~
描述如果在地图中的灰色所标识的平原上部署一支炮兵部队,则图中的黑色的网格表示它能够攻击到的区域:沿横向左右各两格,沿纵向上下各两格。图上其它白色网格均攻击不到。从图上可见炮兵的攻击范围不受地形的影响。
现在,将军们规划如何部署炮兵部队,在防止误伤的前提下(保证任何两支炮兵部队之间不能互相攻击,即任何一支炮兵部队都不在其他支炮兵部队的攻击范围内),在整个地图区域内最多能够摆放多少我军的炮兵部队。
-
输入
-
第一行输出数据测试组数X(0<X<100)
接下来每组测试数据的第一行包含两个由空格分割开的正整数,分别表示N和M; 接下来的N行,每一行含有连续的M个字符('P'或者'H'),中间没有空格。按顺序表示地图中每一行的数据。0<=N <= 100;0<=M <= 10。
输出
-
每组测试数据输出仅一行,包含一个整数K,表示最多能摆放的炮兵部队的数量。
样例输入
-
1 5 4 PHPP PPHH PPPP PHPP PHHP
样例输出
-
6
状态压缩
dp[row][i][j] 当前行row行,row行状态为i, row - 1行状态为j 时最多人数。
#include <iostream>
#include <string>
#include <string.h>
#include <map>
#include <stdio.h>
#include <algorithm>
#include <queue>
#include <vector>
#include <math.h>
#include <set>
#define Max(a,b) ((a)>(b)?(a):(b))
#define Min(a,b) ((a)<(b)?(a):(b))
#define SWAP(a , b){ int temp = a; a = b; b = temp;}
#define inf 100000000
using namespace std;
typedef long long LL;
int N, M;
int cannot[102];//下标从1开始
int dp[102][70][70];
vector<int> states;
vector<int> people;
bool ok(int x)
{
if(x&x<<1)
return false;
if(x&x<<2)
return false;
if(x&x>>1)
return false;
if(x&x>>2)
return false;
return true;
}
int Count(int x)//x状态的人数
{
int t = 0;
while(x)
{
t++;
x &= (x - 1);
}
return t;
}
void get_all_states()//不考虑是不是高山,满足条件的状态
{
states.clear();
people.clear();
for(int i = 0;i < (1<<M);i++)
{
if(ok(i))
{
states.push_back(i);
people.push_back(Count(i));
}
}
}
int DP()
{
int i, j , k , row , ans = 0;
memset(dp , 0 , sizeof(dp));
get_all_states();
for(i = 0;i < states.size();i++)//先考虑第一行
{
if(states[i]&cannot[1])
continue;
else
dp[1][i][0] = people[i];
}
for(int row = 2;row <= N;row++)//从第二行开始
{
for(i = 0;i < states.size();i++)//枚举该行的状态
{
if(states[i]&cannot[row])
continue;
for(j = 0;j < states.size();j++)//枚举上一行的状态
{
if(states[j]&cannot[row - 1])
continue;
if(states[i]&states[j])
continue;
for(k = 0;k < states.size();k++)//枚举上第二行的状态
{
if(states[k]&cannot[row - 2])
continue;
if(states[i]&states[k])
continue;
if(states[j]&states[k])
continue;
dp[row][i][j] = Max(dp[row][i][j] , dp[row - 1][j][k] + people[i]);
}
}
}
}
for(i = 0;i < states.size();i++)
for(j = 0;j < states.size();j++)
ans = Max(dp[N][i][j],ans);
return ans;
}
int main()
{
int t;
cin >> t;
while(t--)
{
memset(cannot , 0 , sizeof(cannot));
scanf("%d%d",&N,&M);
string s;
for(int i = 1;i <= N;i++)
{
cin >> s;
for(int j = 0;j < M;j++)
if(s[j] == 'H')
cannot[i] |= 1<<(M - j - 1) ;
}
cout << DP() << endl;
}
return 0;
}