HDU 1561 The more, The Better (树形dp)

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1561

题意不讲了，中文。

树形背包，可以以0为总root，m++。dp[i][j] 表示以i节点为root 攻克j个城堡的价值最大是多少。 dp[i][j] = max(dp[i][j] , dp[i.son][k] + dp[i][j - k])

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
const int N = 205;
int dp[N][N], n, m, a[N];
vector <int> G[N];

void init(int n) {
    for(int i = 0; i <= n; ++i) {
        G[i].clear();
    }
    memset(dp, 0, sizeof(dp));
}

void dfs(int u, int p) {
    for(int i = 1; i <= m; ++i) {
        dp[u][i] = a[u];
    }
    for(int i = 0; i < G[u].size(); ++i) {
        int v = G[u][i];
        if(v == p)
            continue;
        dfs(v, u);
        for(int j = m; j >= 1; --j) { //类似背包，倒着来取值可以唯一
            for(int k = 1; k < j; ++k) {
                dp[u][j] = max(dp[u][j], dp[v][j - k] + dp[u][k]);
            }
        }
    }
}

int main()
{
    while(~scanf("%d %d", &n, &m) && (n || m)) {
        int u, v;
        init(n);
        for(int i = 1; i <= n; ++i) {
            scanf("%d %d", &u, &v);
            G[u].push_back(i);
            a[i] = v;
        }
        ++m;
        dfs(0, -1);
        printf("%d
", dp[0][m]);
    }
    return 0;
}