树的直径

BFS求树的直径：第一遍bfs求得一定是直径的端点。再做bfs
题目描述
给定一棵树 T ，树 T 上每个点都有一个权值。
定义一颗树的子链的大小为：这个子链上所有结点的权值和 。
请在树 T 中找出一条最大的子链并输出。
输入描述:
5
2 -1 -1 -2 3
1 2
2 3
2 4
2 5
输出
4
大板子一套，转移都省了

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N=2e5+10;
vector<ll> mp[N];
ll arr[N];
ll n,vis[N],dis[N],pos;
void bfs(ll x){
    memset(vis,0,sizeof(vis)); 
    memset(dis,0,sizeof(dis));
    pos=x;
    vis[x]=1,dis[x]=arr[x];
    queue<ll> q; 
    q.push(x);
    while(!q.empty()){
        ll u=q.front();q.pop();
        for(ll i=0;i<mp[u].size();i++){
            if(!vis[mp[u][i]]){
                vis[mp[u][i]]=1;
                dis[mp[u][i]]=dis[u]+arr[mp[u][i]];
                q.push(mp[u][i]);
                if(dis[mp[u][i]]>dis[pos]) pos=mp[u][i];
            }
        }
    }
}
int main(){
    ll u,v,a,b;
    scanf("%lld",&n);
    for(ll i=1;i<=n;i++){
        scanf("%lld",arr+i);
    }
    for(ll i=1;i<n;i++){//建图
        scanf("%lld%d",&u,&v); mp[u].push_back(v); mp[v].push_back(u);
    }
    bfs(1); a=pos;  bfs(pos);   b=pos;
    cout<<dis[b]<<endl;
    return 0;
}