[CodeForces-1084D][补题][树形DP(裸)]

题目是说有很多个城市，到达每个城市都可以获得相应的燃油，然后去往另外一个城市要消耗相应的燃油，不限制起点或者终点，怎么样才能获得最多的燃油？

题目第一行先输入一个n代表城市个数，然后接下来一行代表每个城市的获得燃油量，然后n-1行的路径和耗油量。

这道题使用树形DP算法，dfs搜索，一开始比较难以理解的就是怎么实现截取中间的一段路，因为dfs是从起始点一路向下搜索，后来发现，每次访问到一个点，先用这个城市的获得燃油量初始化dp数组，然后后面在更新dp的时候，比较子树+这个点的获得燃油量 - 消耗量（即拓展子树）和这个点初始化的权值，就可以决定是否保留子树。

注意，树形dp是使用dfs直接搜索到底，然后从底部开始向上dp，子树就是子问题，然后在子问题的基础上拓展，求出最优解。

#include <bits/stdc++.h>
#define maxn 300009
#define ll long long

using namespace std;

ll v[maxn];
vector<pair<ll,ll> >node[maxn];
int n;
ll dp[maxn]; 
ll ans = 0;

void dfs(int now,int pre){
	dp[now] = v[now];
	ans = max(dp[now],ans);
	for(int i = 0;i < node[now].size();i++){
		int next = node[now][i].first;
		if(next == pre){
			continue;
		}
		dfs(next,now);
		ans = max(dp[now] + dp[next] - node[now][i].second,ans);
		dp[now] = max(dp[now],dp[next] + v[now] - node[now][i].second);
	}
}

int main(){
	scanf("%d",&n);
	for(int i = 1;i <= n;i++){
		scanf("%lld",&v[i]);
	}
	ll a,b,c;
	for(int i = 0;i < n - 1;i++){
		scanf("%lld %lld %lld",&a,&b,&c);
		node[a].push_back(make_pair(b,c));
		node[b].push_back(make_pair(a,c));
	}
	dfs(1,0);
	printf("%lld\n",ans);
	return 0;
}