树的直径 树的路径权值和

树直径的模板

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
const int N=1e5+9;
int n,head[N],to[N<<1],nxt[N<<1],cnt=0;
ll dp[N][2];
ll a[N];
ll ans=0;

inline void add(int x,int y)
{
  nxt[++cnt]=head[x];
  head[x]=cnt;
  to[cnt]=y;
  return ;
}
//这个 ‘次远距离’ 并不是真正的从 x 到最底端的次远，
//而是再用 x 的其他子节点来更新，所找到的最远距离
//dp[i][0]、dp[i][1]表示i结点的来自不同结点的最长链的长度
//对于一个子结点，其分支分别存到了最长和次长当中去
//他的兄弟结点的两个子链与已经存储的值进行比较，相当于4选2，这样就能够存最大和最小了

void dfs(int x,int fa)
{//to表示的是子结点
 //nxt表示的是兄弟结点
  for(int i=head[x];i;i=nxt[i])
  {
    int y=to[i];
    if(y==fa) continue;
    dfs(y,x);//检查所有子结点
    if((dp[y][0]+a[y])>dp[x][0])
    {
      dp[x][1]=dp[x][0];
      dp[x][0]=dp[y][0]+a[y];
    }
    else if((dp[y][0]+a[y])>dp[x][1])
    {
      dp[x][1]=dp[y][0]+a[y];
    }
  }
  return ;
}

int main()
{
  cin>>n;
  for(int i=2;i<=n;i++)//??
  {
    int x,y;cin>>x>>y;
    add(x,y);add(y,x);
  }
  for(int i=1;i<=n;i++)cin>>a[i];
  dfs(1,0);
  for(int i=1;i<=n;i++) ans=max(ans,dp[i][0]+dp[i][1]+a[i]);//两个不同子树的最大长度
  cout<<ans;
  return 0;
}