树上子链 （树上直径模板）

我们假设树的根是rt
那么树的直径显然会有两种情况
1.直径的两个端点分别在rt的两个子树内
2.直径在rt的某个子树内
这里就可以用到分治的思想，先以rt为根处理情况1，再递归rt的子树将情况2转变为情况1处理

#include<bits/stdc++.h>
#define ll long long 
using namespace std;
int a[100005];
vector<int> G[100005];
int n;
ll ans=-1e16;
ll dp[100005];//以i为根节点,到叶子节点的最大路径 
void dfs(int now,int fa)
{
	ll max1=a[now],max2=a[now];//max1 到叶子节点的最大路径 max2次大路径 
	for(auto t:G[now])
	{
		if(t==fa) continue;
		dfs(t,now);
		if(dp[t]+a[now]>max1)//维护max1 
		{
			max2=max1;
			max1=dp[t]+a[now];
		}
		else if(dp[t]+a[now]>max2)//维护max2
		{
			max2=dp[t]+a[now];
		}		
	}
	dp[now]=max1;
	ans=max(ans,max1+max2-a[now]);	
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
	} 
	for(int i=1;i<n;i++)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		G[u].push_back(v);
		G[v].push_back(u);
	}
	dfs(1,0);
	printf("%lld\n",ans);
}