NYOJ 81 炮兵阵地 （状态压缩）

OJ题目 ： 戳这里~~

描述

司令部的将军们打算在N*M的网格地图上部署他们的炮兵部队。一个N*M的地图由N行M列组成，地图的每一格可能是山地（用"H" 表示），也可能是平原（用"P"表示），如下图。在每一格平原地形上最多可以布置一支炮兵部队（山地上不能够部署炮兵部队）；一支炮兵部队在地图上的攻击范围如图中黑色区域所示： 

如果在地图中的灰色所标识的平原上部署一支炮兵部队，则图中的黑色的网格表示它能够攻击到的区域：沿横向左右各两格，沿纵向上下各两格。图上其它白色网格均攻击不到。从图上可见炮兵的攻击范围不受地形的影响。 
现在，将军们规划如何部署炮兵部队，在防止误伤的前提下（保证任何两支炮兵部队之间不能互相攻击，即任何一支炮兵部队都不在其他支炮兵部队的攻击范围内），在整个地图区域内最多能够摆放多少我军的炮兵部队。

输入

第一行输出数据测试组数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;
}