hdu 1011 Starship Troopers 树形DP

难度：6

题型：树形DP

一棵树n个结点，每个结点上有两个值，分别表示bugs数量和价值。现在有p个士兵，从结点1开始攻击，一个士兵能攻击20个bugs，把某个结点的bugs都攻击完才能得到结点上的价值，一个士兵只能攻击一个结点，也就是士兵停留在攻击的结点上，走过的结点不能再走，走到一个结点上必须攻击完所有的bugs，否则不能走。问能获得的最大价值。

状态转移方程dp[u][j] = max(dp[u][j] , dp[u][j-k] + dp[v][k]) , 其中v是u的儿子节点

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 110;
int n , m , dp[maxn][maxn] , bugs[maxn] , val[maxn];
bool vis[maxn];
struct Edge {
    int v , next;
    Edge() {}
    Edge(int v , int next) : v(v) , next(next) {}
}edge[maxn<<1];
int E , head[maxn];
void init() {
    E = 0; memset(head,-1,sizeof(int)*(n+1));
}
void addedge(int u,int v) {
    edge[E] = Edge(v , head[u]); head[u] = E++;
    edge[E] = Edge(u , head[v]); head[v] = E++;
}
void dfs(int u) {
    if(vis[u]) return;
    vis[u] = true;
    int cost = (bugs[u] + 19) / 20;
    for(int i=cost;i<=m;i++) dp[u][i] = val[u];
    for(int i=head[u];i!=-1;i=edge[i].next) {
        int v = edge[i].v;
        if(vis[v]) continue;
        dfs(v);
        for(int j=m;j>=cost;j--) {
            for(int k=0;j-k>=cost;k++) {
                dp[u][j] = max(dp[u][j] , dp[u][j-k] + dp[v][k]);
            }
        }
    }
    if(dp[u][0] > 1) {
        dp[u][1] = max(dp[u][1] , dp[u][0]);
        dp[u][0] = 0;
    }
    return;
}
int main() {
    while(~scanf("%d%d" , &n , &m) && n != -1) {
        init();
        memset(vis , false , sizeof(vis));
        memset(dp , 0 , sizeof(dp));
        for(int i=1;i<=n;i++) scanf("%d%d" , &bugs[i] , &val[i]);
        for(int i=1;i<n;i++) {
            int u , v;
            scanf("%d%d" , &u , &v);
            addedge(u , v);
        }
        dfs(1);
        int ans = 0;
        for(int i=1;i<=m;i++) if(dp[1][i] > ans) ans = dp[1][i];
        printf("%d\n" , ans);
    }
    return 0;
}