树上子链 （树型dp求树的直径）

题目链接

题意：

思路：首先把无根树转化成有根树，然后设置dp方程，dp【i】为以i为根节点的包括i在内的最长链的长度。dp【u】=dp【v】+val【u】我们还要保存次长链的长度。那么最终的答案就是，枚举所有节点，把这个节点作为根，最长链加上次长链的长度就是最终的答案，代码如下。

#include <bits/stdc++.h>
#define pb push_back
#define ll long long
using namespace std;
const int maxn = 1e5+100;
const ll inf = 1e11+10;
int val[maxn];
ll dp[maxn];
 
vector<int>G[maxn];
 
ll ans = -inf;
void dfs(int u,int fa)
{
    ll tmp1,tmp;
    tmp1 = tmp = val[u];
    for(int i=0;i<G[u].size();i++)
    {
        int v = G[u][i];
        if(v == fa)continue;
        dfs(v,u);
        if(dp[v] + val[u] > tmp)
        {
            tmp1 = tmp;
            tmp = dp[v] + val[u];
        }
        else if(dp[v] + val[u] > tmp1)
        {
            tmp1 = dp[v] + val[u];
        }
    }
    dp[u] = tmp;
    ans = max(tmp1+tmp-val[u] , ans);
}
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1; i<=n;i++)scanf("%d",val+i);
    for(int i=1;i<n;i++)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        G[u].pb(v);
        G[v].pb(u);
    }
    dfs(1,-1);
    printf("%lld\n",ans);
}