Codeforces Round #526 (Div. 2)D. The Fair Nut and the Best Path(树形DP)

题目链接：https://codeforces.com/contest/1084/problem/D

 

题目大意：可以从一个点得到x的燃料，然后花权值的燃料到下一个点，问任意两点可以得到的最大燃料是多少

 

题目思路：dfs，对于每一个点如果当它是根时，他的最大燃料就是自己作为中转站，儿子中最大和次大的相加加上本地的燃料。还有一种可能是父亲以该点为中转站，此时就只用该点的最大值往上传，父亲选用它时他就是作为中转站使用。所以dp只用记录儿子中最大的即可。

 

以下是代码：

#include<bits/stdc++.h>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<math.h>
using namespace std;
#define inf 0x3f3f3f3f
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define per(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
const int MAXN = 3e5+5;
const int MOD = 1e9+7;
struct Edge{
    ll to,c;
};
vector<Edge>v[MAXN];
ll a[MAXN];
ll dp[MAXN],dp_z[MAXN],ans=0;
void dfs(int u, int fa){
    ll mx=0,mx2=0;
    int len=v[u].size();
    rep(i,0,len-1){
        int to=v[u][i].to,w=v[u][i].c;
        if(to==fa)continue;
        dfs(to,u);
        if(dp[to]-w>mx){
            mx2=mx;
            mx=dp[to]-w;
        }
        else if(dp[to]-w>mx2)mx2=dp[to]-w;
    }
    dp_z[u]=mx+mx2+a[u];
    dp[u]=mx+a[u];
    ans=max(dp_z[u],ans);
}
int main(){
    int n,x,y,z;
    while(~scanf("%d",&n)){
        memset(dp,0,sizeof(dp));
        memset(dp_z,0,sizeof(dp_z));
        rep(i,1,n)scanf("%I64d",&a[i]),v[i].clear();
        rep(i,1,n-1){
            scanf("%d%d%d",&x,&y,&z);
            v[x].push_back(Edge{y,z});
            v[y].push_back(Edge{x,z});
        }
        dfs(1,-1);
        printf("%I64d\n",ans);
    }
    return 0;
}