题意:给定一棵树以及边的权值,边的权值表示从一个顶点到达另一个顶点的用时,每个顶点有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;
}