点对最大值——简单树形dp

题目链接：https://ac.nowcoder.com/acm/contest/5757/A
时间限制：C/C++ 1秒，其他语言2秒
空间限制：C/C++ 131072K，其他语言262144K
64bit IO Format: %lld
题目描述
这里有一棵树，每个点和每条边都存在一个价值。对于树上点对的价值，包括点对的起点和终点以及路径上边权值之和，不包括路径上其他点值。
求这颗树上最大的点对价值为多少。点对至少需要两个点。

输入描述:
输入t，代表有t组样例。每组样例第一行输入n，代表有n个点。接下来有n-1行，第i行有a[i]和b[i]，代表a[i]节点与i节点存在一条边，且边的值为b[i]，2<=i<=n。接下来一行有n个值c[j]，代表每个节点j的价值，1<=j<=n。
（t<=10，n>1，n<1e6，a[i]<i，-500<=b[i]<=500，-500<=c[j]<=500）
输出描述:
输出最大的点对价值

示例1
输入
1
4
1 -2
1 2
1 3
2 -2 3 4

输出
12

思路：
我们可以维护 dp[i]: 以i为根的子树中最长的链（不包括节点i的值）。
对于任意结点 u，我们需要子树中 最长的链mmx 和 次大的链mx
那么我们可以每次处理完节点 u 的子树后，更新 ans=mmx+mx+dp[u]，然后更新 dp[u]=mmx。
由于避免 dp[i] 所表示的链断开而错误，我们在初始化 mmx=i结点的val，这样就可以保证，对于子树中的dp[i]表示的一定是和 i 结点连续的链。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define ull unsigned long long
const int maxn=1e6+7;
const int inf=0x3f3f3f3f;
///std::ios::sync_with_stdio(false);
struct tr
{
    int to;
    int val;
//    tr(int t, int v) : to(t), val(v) {}
};

vector<tr>g[maxn];
int dp[maxn];
int ans;
void add(int u,int v,int val)
{
    tr x;
    x.to=u;
    x.val=val;
    g[v].push_back(x);
    x.to=v;
    g[u].push_back(x);
//    g[u].push_back(tr(v,val));
//    g[v].push_back(tr(u,val));
}

void dfs(int u,int fa)
{
    int mmx=dp[u];
    int mx=-inf;
    for(int i=0;i<g[u].size();i++)
    {
        if(g[u][i].to==fa)
            continue;
        int v=g[u][i].to;
        int val=g[u][i].val;
        dfs(v,u);
        mx=max(mx,val+dp[v]);
        if(mx>mmx)
        {
            swap(mmx,mx);
        }
        ans=max(ans,mx+mmx);
        dp[u]=mmx;
    }
}
int main()
{
    std::ios::sync_with_stdio(false);
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n;
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            g[i].clear();
        }
        ans=-inf;
        for(int i=2;i<=n;i++)
        {
            int a,b;
            scanf("%d%d",&a,&b);
            add(i,a,b);

        }
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&dp[i]);
        }
        dfs(1,-1);
        printf("%d\n",ans);
    }
}