哈尔滨理工大学第十届程序设计竞赛-点对最大值——（树形DP——最大简单路径权值和改版）

总结

以前写了一个裸体，直径和的裸体，可能只是把代码对着敲了一遍吧，理解的不是很深，一到比赛改版题，就完全懵逼了，裸体原理忘了，改版的更别说能不能写。

算法核心

出发点：点对至少需要两个点。
枚举每条边的u和v，u是v父节点，v向下的最大直径+cost+u向上的最大直径

向下的最大直径，就是看最大跟次大，但是每次先加入当前点值，保证每次直径存在尾点权

向上的最大直径，u结点先更新点权跟向上最大简单路径，再更新v的最大直径点权和

题目链接

//#pragma GCC optimize(2)
//#pragma GCC target ("sse4")
#include<bits/stdc++.h>
//typedef long long ll;
#define ull       unsigned long long
#define int       long long
#define F           first
#define S           second
#define endl        "\n"//<<flush
#define eps         1e-6
#define base        131
#define lowbit(x)   (x&(-x))
#define PI          acos(-1.0)
#define inf         0x3f3f3f3f
#define MAXN        0x7fffffff
#define INF         0x3f3f3f3f3f3f3f3f
#define ferma(a,b)  pow(a,b-2)
#define mod(x)      (x%mod+mod)%mod
#define pb          push_back
#define decimal(x)  cout << fixed << setprecision(x);
#define all(x)      x.begin(),x.end()
#define memset(a,b) memset(a,b,sizeof(a));
#define IOS         ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
using namespace std;
void file()
{
#ifdef ONLINE_JUDGE
#else
    freopen("D:/LSNU/codeforces/duipai/data.txt","r",stdin);
    //  freopen("D:/LSNU/codeforces/duipai/WA.txt","w",stdout);
#endif
}
const int N=1e6+5;
vector<pair<int,int> >G[N];
int dp[N][3],a[N],tag[N];
void dfs(int u,int fat)
{
    dp[u][0]=a[u];
    for(auto it:G[u])
    {
        int v=it.F,cost=it.S;
        if(v==fat)
            continue;
        dfs(v,u);
        if(dp[u][0]<=dp[v][0]+cost)
        {
            dp[u][1]=dp[u][0];
            dp[u][0]=dp[v][0]+cost;
            tag[u]=v;
        }
        else if(dp[u][1]<dp[v][0]+cost)
            dp[u][1]=dp[v][0]+cost;
    }
}
int ans;
void dfs1(int u,int fat)
{
    dp[u][2]=max(dp[u][2],a[u]);
    for(auto it:G[u])
    {
        int v=it.F,cost=it.S;
        if(v==fat)
            continue;
        if(tag[u]==v)
        {
            dp[v][2]=max(dp[u][1],dp[u][2])+cost;
            ans=max(ans,max(dp[u][2],dp[u][1])+dp[v][0]+cost);
        }
        else
        {
            dp[v][2]=max(dp[u][0],dp[u][2])+cost;
            ans=max(ans,max(dp[u][2],dp[u][0])+dp[v][0]+cost);
        }
        dfs1(v,u);

    }
}
signed main()
{
    IOS;
    file();
    int t;
    cin>>t;
    while(t--)
    {
        int n;
        cin>>n;
        memset(dp,-inf);
        memset(tag,0);
        for(int i=1;i<=n;i++)
            G[i].clear();
        for(int u=2;u<=n;u++)
        {
            int v,cost;
            cin>>v>>cost;
            G[u].pb({v,cost});
            G[v].pb({u,cost});
        }
        for(int i=1;i<=n;i++)
            cin>>a[i];
        ans=-INF;
        dfs(1,0);
        dfs1(1,0);
        cout<<ans<<endl;
    }





    return 0;
}