ZOJ 3626 —— Treasure Hunt I（树形DP + 背包）

题目：Treasure Hunt I

题意：给定一棵树以及边的权值，边的权值表示从一个顶点到达另一个顶点的用时，每个顶点有v[i]价值的宝物，现在要从root出发，最后回到root并且时间不超过m，求得到的最大价值。

还是树形dp，从root开始dfs，对物品价值做背包，转移的时候记得边是一去一回，所以用时要乘以2。

dp[i][j]表示从i出发再回到i（不包含其父节点，由root出发的dfs决定父子关系），用时不超过j的最大价值。

#include<cstdio>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
const int N = 110;
const int M = 1100;
#define pb push_back
vector<int> V[N];
vector<int> G[N];
int n, m, v[N], dp[N][M];
void dfs(int x, int fa){
    for(int i=0; i<=m; i++) dp[x][i]=v[x];
    for(int i=0; i<V[x].size(); i++){
        int j = V[x][i];
        if(j==fa)   continue;
        int d = G[x][i] * 2;
        dfs(j, x);
        for(int k=m; k>=d; k--){
            for(int p=0; p+d<=k; p++)   dp[x][k] = max(dp[x][k], dp[x][k-d-p]+dp[j][p]);
        }
    }
}
int main(){
    int root;
    while(~scanf("%d", &n)){
        for(int i=1; i<=n; i++){
            V[i].clear();
            G[i].clear();
            scanf("%d", v+i);
        }
        int a,b,c;
        for(int i=1; i<n; i++){
            scanf("%d %d %d", &a, &b, &c);
            V[a].pb(b); G[a].pb(c);
            V[b].pb(a); G[b].pb(c);
        }
        scanf("%d %d", &root, &m);
        dfs(root, -1);
        printf("%d\n", dp[root][m]);
    }
    return 0;
}