哈爾濱理工大學第十屆程序設計競賽-點對最大值——（樹形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;
}