树上子链-树形DP

给定一棵树 T ，树 T 上每个点都有一个权值（有负值）。
定义一颗树的子链的大小为：这个子链上所有结点的权值和 。
请在树 T 中找出一条最大的子链并输出。

输入描述:
第一行输入一个 n,1 \le n \le 10^5n,1≤n≤10
5
。
接下来一行包含n个数，对于每个数 a_i, -10^5 \le a_i \le 10^5a i,−10 5≤a i ≤10 5
，表示 i 结点的权值。
接下来有n-1行，每一行包含两个数u，v（ 1 \le u,v \le n1≤u,v≤n , u != v），表示u与v之间有一条边。
输出描述:
仅包含一个数，表示我们所需要的答案。

树形DP模板

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const ll maxn=1e6+5;
ll dp[maxn],a[maxn],res;
vector<ll>vec[maxn];
void dfs(ll u,ll fa)
{
    dp[u]=a[u];
    ll len=vec[u].size();
    for(ll i=0;i<len;i++)
    {
        ll v=vec[u][i];
        if(v!=fa)
        {
            dfs(v,u);
            res=max(res,dp[u]+dp[v]);
            dp[u]=max(dp[u],dp[v]+a[u]);
        }
    }
    res=max(res,dp[u]);
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    ll n;
    cin>>n;
    for(ll i=1; i<=n; i++)
        cin>>a[i];
    ll u,v;
    for(ll i=1; i<n; i++)
    {
        cin>>u>>v;
        vec[u].push_back(v);
        vec[v].push_back(u);
    }
    res=-199999999999999999;
    dfs(1,0);
    cout<<res<<endl;
}