POJ 1185 炮兵阵地(状态压缩DP)

题意:

求最大的炮兵摆放数量,并且使他们之间不相互误伤。

NOI 01 的题目。

思路:

1. 和 POJ 1038 类似,把行看成是一个整体,每一种摆放代表着一种状态,r-2, r-1 行的状态决定着 r 行的状态;

2. 本题状态比较稀疏,所以提前处理下炮兵的摆放无意是比较好的选择,由于没两个炮兵之间至少相距为 2,利用这个把每一行的炮兵可能摆放情况存放起来;

3. dp[r][i][j] 表示第 r 行的状态为 i 且第 r-1 行的状态为 j 时,所能摆放的最大炮兵数量,同时还需要枚举 r-2 行的炮兵摆放情况;

4. 最终有状态转移方程: dp[r][i][j] = max(dp[r][i][j], dp[r-1][j][k] + men[i]); 并且要求 dp[r-1][j][k] 存在,且 r 行能保证能摆放 i 状态;

5. 有了 POJ 1038 和 POJ 1185 的积累,我想以后做此类基于摆放的状态压缩 DP 应该能够有个比较清晰的思路了。

#include <iostream>
#include <algorithm>
using namespace std;

int row, col;
int state[60], cnt, men[60], mapstate[110];
bool grid[110][12];
int dp[2][60][60];

bool judge(int s) {
    if (s & (s<<1))
        return false;
    if (s & (s<<2))
        return false;
    return true;
}

int calcmen(int s) {
    int ans = 0;
    while (s) {
        ans += 1;
        s &= (s-1);
    }
    return ans;
}

void initstate(int endstate) {
    cnt = 0;
    for (int s = 0; s < endstate; s++) {
        if (judge(s)) {
            state[cnt] = s;
            men[cnt] = calcmen(s);
            cnt += 1;
        }
    }
}

int solvedp() {
    memset(dp[0], -1, sizeof(dp[0]));
    for (int i = 0; i < cnt; i++) {
        if (mapstate[0] & state[i])
            continue;
        dp[0][i][0] = men[i];
    }
    int T1 = 1, T2 = 0;
    for (int r = 1; r < row; r++) {
        T1 ^= 1, T2 ^= 1;
        memset(dp[T2], -1, sizeof(dp[0]));
        for (int i = 0; i < cnt; i++) {
            if (state[i] & mapstate[r]) 
                continue;
            for (int j = 0; j < cnt; j++) {
                if ((state[i] & state[j]) || (state[j] & mapstate[r-1]))
                    continue;
                for (int k = 0; k < cnt; k++) {
                    if ((state[i] & state[k]) || (state[k] & state[j]))
                        continue;
                    if (dp[T1][j][k] != -1)
                        dp[T2][i][j] = max(dp[T2][i][j], dp[T1][j][k]+men[i]);
                }
            }
        }
    }
    int ans = -1;
    for (int i = 0; i < cnt; i++)
        for (int j = 0; j < cnt; j++)
            ans = max(ans, dp[T2][i][j]);
    return ans;
}

int main() {
    scanf("%d%d", &row, &col);
    for (int i = 0; i < row; i++) {
        char s[20];
        scanf("%s", s);
        mapstate[i] = 0;
        for (int j = 0; j < col; j++) {
            mapstate[i] <<= 1;
            mapstate[i] |= s[j] == 'H' ? 1 : 0;
        }
    }
    initstate(1<<col);
    printf("%d\n", solvedp());
    return 0;
}
-------------------------------------------------------

kedebug

Department of Computer Science and Engineering,

Shanghai Jiao Tong University

E-mail: kedebug0@gmail.com

GitHub: http://github.com/kedebug

-------------------------------------------------------

原文地址:https://www.cnblogs.com/kedebug/p/3001642.html