3760. 最大剩余油量 (树形dp)

题目链接

题意：

一个国家由 nn 个城市组成，这 n 个城市由 n−1 条双向道路连接，呈一个树形结构。

每个城市都设有加油站，在第 ii 个城市可以购买 wi 升汽油。

汽车在道路上行驶，毫无疑问也会消耗汽油，每条道路的具体耗油量也会给出。

现在，需要制定一条汽车的行进路线，从任意城市 s 出发，经过一条简单路径，到达任意城市 e 结束。

注意，行进路线也可以只包含一个城市（也就是哪都没去）。

汽车初始时油箱是空的，但是可以在路线中经过的每个城市购买汽油，包括开始城市和最终城市。

如果在一条行进路线中，汽车沿一条道路从某一城市开往另一城市时，现有油量小于该条道路所需油量，那么就说明这条行进路线行不通。

请问，在保证行进路线合理的情况下，汽车在抵达最终城市后，可以剩余的最大油量是多少？

再次提醒，汽车在最终城市也可以加油。

思路：由于每个点都可以作为起点，所以答案应该在dfs的过程中进行更新，每个点找一个最大值和第二大值去更新，然后返回一个最大值和当前权值到父节点。

代码：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
int p[300010];
ll ans=0;
vector<pair<int,int> >w[300010];
ll dfs(int x,int fa){
    int k=w[x].size();
    ll d1=0,d2=0;
    for(int i=0;i<k;i++){
        int v=w[x][i].first;
        if(v==fa)continue;
        ll f=dfs(v,x);
        if(f<w[x][i].second)continue;
        f-=w[x][i].second;
        if (f>= d1) d2 = d1, d1 = f;
        else if (f > d2) d2 = f;
    }
    ans = max(ans, d1 + d2 + p[x]);
    return d1 + p[x];
}
int main(){
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&p[i]);
    }
    int a,b,c;
    for(int i=1;i<n;i++){
        scanf("%d%d%d",&a,&b,&c);
        w[a].push_back(make_pair(b,c));
        w[b].push_back(make_pair(a,c));
    }
    ans=max(ans,dfs(1,-1));
    printf("%lld\n",ans);
    return 0;
}