[POJ]1185 炮兵阵地 [状压dp]

Description

司令部的将军们打算在N*M的网格地图上部署他们的炮兵部队。一个N*M的地图由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<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;

const int MAXN = 105;
int fib[MAXN], ste[1<<6], num[1<<6], sz; // 每行最多60个状态
int dp[MAXN][1<<6][1<<6], n, m;

int __num(int x) {
    int cnt = 0;
    while(x) {
        if(x&1) ++cnt;
        x >>= 1;
    }
    return cnt;
}

void slove() {
    for(int i = 0; i < 1<<m; ++i) {
        if(i&(i<<1|i<<2)) continue;
        num[sz] = __num(i);
        if(!(i&fib[1])) dp[1][0][sz] = num[sz];
        ste[sz++] = i;
    }
    for(int i = 0; i < sz; ++i) {
        if(ste[i]&fib[2]) continue;
        for(int j = 0; j < sz; ++j) {
            if(ste[i]&ste[j] || ste[j]&fib[1]) continue;
            dp[2][j][i] = dp[1][0][j] + num[i];
        }
    }
    for(int i = 3; i <= n; ++i) {
        for(int j = 0; j < sz; ++j) {
            if(ste[j]&fib[i]) continue;
            for(int k = 0; k < sz; ++k) {
                if(ste[j]&ste[k] || ste[k]&fib[i - 1]) continue;
                for(int z = 0; z < sz; ++z) {
                    if(ste[j]&ste[z] || ste[k]&ste[z] || ste[z]&fib[i - 2]) continue;
                    dp[i][k][j] = max(dp[i][k][j], dp[i - 1][z][k] + num[j]);
                }
            }
        }
    }
    int ans = 0;
    for(int i = 1; i <= n; ++i) {
        for(int j = 0; j < sz; ++j) {
            if(fib[i]&ste[j]) continue;
            for(int k = 0; k < sz; ++k) {
                ans = max(ans, dp[i][k][j]);
            }
        }
    }
    printf("%d\n", ans);
}

int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; ++i) {
        getchar();
        for(int j = 0; j < m; ++j) {
            if(getchar() == 'H') fib[i] += 1<<j;
        }
    }
    slove();
    return 0;
}