树的直径模板

struct stu
{
    int from,to,val,next;
}st[M];                        //定义结构体储存节点和节点间的距离以及下一个节点的位置 


void init()
{
    num=0;
    memset(head,-1,sizeof(head));
}                            //初始化函数 


void add_edge(int u,int v,int w)
{
    st[num].from=u;                //储存节点u 
    st[num].to=v;                //储存节点v 
    st[num].val=w;                //储存节点u与v的距离 
    st[num].next=head[u];        //储存下一个与节点u相连的节点的位置 
    head[u]=num++;                //储存u节点的位置 
}                            


void bfs(int fir)                //搜索函数 
{    
    ans=0;
    int u;
    memset(sum,0,sizeof(sum));    //以该节点结尾的最长路 
    memset(flag,0,sizeof(flag));//标记被访问的节点 
    queue<int>que;
    que.push(fir);
    flag[fir]=1;
    while(!que.empty())
    {
    
        u=que.front();
        que.pop();
        for(i = head[u] ; i != -1 ; i=st[i].next)
        {
            if(!flag[st[i].to] && sum[st[i].to] < sum[u]+st[i].val)
            {
                sum[st[i].to]=sum[u]+st[i].val;
                if(ans < sum[st[i].to])
                {
                    ans=sum[st[i].to];                //记录最大距离 
                    node=st[i].to;                    //记录端点 
                }
                flag[st[i].to]=1;
                que.push(st[i].to);
            }
        }
    }
}

假设从节点1开始，找到节点u使得1到u的距离最大，那么节点u到树上其他节点的最大距离就是树的直径（树上两点间的最大距离） 

例题链接