hdu 1011 Starship Troopers 树形dp

http://acm.hdu.edu.cn/showproblem.php?pid=1011
- 题意
- 给 n 个点, m 个士兵
- n 个点由 n-1 条边连接为树
- 根节点为点 1
- 每个点内有 a 个 bugs, b 个brains
- 杀死当前点所有的 bugs 才能获得当前点的 brains
- 杀死所有的 bugs 才能移动到子节点.
- 1 个士兵能杀死 20 个bugs
- 不足 20 个 bugs 也需要 1 个士兵.

• 分析
  
  • dfs 搜索所有点
  • i 点的最优
    dp[i][cost] = max(dp[i][cost], dp[son_of_i][k] + dp[i][cost-k]);
  • 特判士兵数量为 0

#include <stdio.h>
#include <iostream>
#include <cstring>
#include <ctime>
#define nr room[now]
using namespace std;
const int MAX = 111;
int n, m, tmp, x, y, mp[MAX][MAX], dp[MAX][MAX];
struct Room {
    int cost, val;
    int depth;
}room[MAX];
int mmax(int a, int b) {
    return a > b ? a : b;
}
void dt(int now, int depth) {
    for (int i = 1; i <= n; ++i)
        if (mp[now][i] && (room[i].depth == 0)) {
            room[i].depth = (depth+1);
            dt(i, depth+1);
        }
}
void getin() {
    memset (room, 0, sizeof(room));
    memset (dp, 0, sizeof(dp));
    memset (mp, 0, sizeof(mp));
    for (int i = 1; i <= n; ++i) {
        scanf("%d%d", &tmp, &room[i].val);
        room[i].cost = (tmp+19)/20;
    }
    for (int i = 0; i < n-1; ++i) {
        scanf("%d%d", &x, &y);
        mp[x][y] = 1, mp[y][x] = 1;
    }
    room[1].depth = 1;
    dt(1, 1);
}
int dfs(int now) {
    for (int i = room[now].cost; i <= m; ++i)
        dp[now][i] = room[now].val;
    for (int i = 1; i <= n; ++i) {
        if (mp[now][i] && room[now].depth < room[i].depth) {
            dfs(i);
            for (int j = m; j >= room[now].cost; --j)
            //此处如果改为 rep(room[now], m, ++) 则wa
            //重复 dp[now][j-k] 的最优解可能会包含 dp[i][k]
            //此时 dp[now][j] 可能会利用多次 dp[i][k]
                for (int k = 1; j-k >= room[now].cost; ++k) {
                    dp[now][j] = mmax(dp[now][j], 
                                dp[now][j-k] + dp[i][k]);
                }
        }
    }
    return dp[now][m];
}
int main() {
//    freopen("in.t", "r", stdin);
//    freopen("out.t", "w", stdout);
    while (scanf("%d%d", &n, &m) == 2 && (n != -1 && m != -1)) {
        getin();
        if (m == 0) {
            puts("0");
            continue;
        }
        dfs(1);
        printf("%d\n", dp[1][m]);
    }
    return 0;
}
/**
判断是否为其子节点而不是其父节点的时候使用了depth
不可否认写挫了
**/

2017-09-04