POJ 1185 状态压缩DP

状态压缩DP

题意：

司令部的将军们打算在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

思路：

​ 状态压缩DP 就是把题中的解转化为数字或者其它状态，常用的是二进制，因为每一个数字都能用二进制表示，但是状态压缩大部分对数量级很敏感，数据量不能太大，而此题就是数据范围小，所以考虑把棋盘的每一行转化为二进制状态，山地不能放兵所以标记为1，放了士兵之后会影响上下左右距离为2的位置，对于左右做出预处理，把所有情况保存起来，对于上下需要枚举，当第一层状态和第二层状态确定之后其它都可以确定。

​ 定义：$dp[r][i][j]$ 为r行的状态是第i种状态，r-1 行状态为第j种。

当把所有情况都保存的时候，同时也需要保存每一状态的士兵数量，然后对于r行，依次按照条件即可，不满足的跳过。

#include <iostream>
#include <cstdio>
#include <cstring>

using namespace std;

int n,m;
int map[105];
int num[70];
int state[70];
int dp[105][70][70];

int main(int argc, char const *argv[])
{
    // freopen("in.txt","r",stdin);

    scanf("%d%d",&n,&m);
    getchar();
    for(int i = 0;i < n; i++) {
        char c;
        for(int j = 0;j < m; j++) {
            scanf("%c",&c);
            if(c == 'H') map[i] += (1<<j);
        }
        getchar();
    }
    int pos = 0;
    for(int i = 0;i < (1<<m); i++) {
        if((i&(i<<1) || (i&(i<<2)))) continue;
        state[pos] = i;
        int k = i;
        int Number = 0;
        while(k) {
            if(k&1) Number++;
            k >>= 1;
        }
        num[pos++] = Number;
    }
    for(int i = 0;i < pos; i++) {
        if(state[i]&map[0]) continue;
        dp[0][i][0] = num[i];
    }
    for(int i = 0;i < pos; i++) {
        if(state[i]&map[1]) continue;
        for(int j = 0;j < pos; j++) {
            if(state[j]&map[0]) continue;
            if(state[i]&state[j]) continue;
            dp[1][i][j] = max(dp[1][i][j],dp[0][j][0] + num[i]);
        }
    }
    for(int r = 2;r < n; r++) {
        for(int i = 0;i < pos; i++) {
            if(state[i]&map[r]) continue;
            for(int j = 0;j < pos; j++) {
                if(state[j]&map[r-1]) continue;
                if(state[j]&state[i]) continue;
                for(int k = 0;k < pos; k++) {
                    if(state[k]&map[r-2]) continue;
                    if(state[k]&state[j]) continue;
                    if(state[k]&state[i]) continue;
                    dp[r][i][j] = max(dp[r][i][j],dp[r-1][j][k] + num[i]);
                }
            }
        }
    }
    int ans = 0;
    for(int i = 0;i < pos; i++) {
        for(int j = 0;j < pos; j++) {
            ans = max(ans,dp[n-1][i][j]);
        }
    }
    printf("%d\n",ans);
    return 0;
}