计蒜客习题 网络延时 （树的直径）

问题描述

，或者通过其他网络间接连通。 为了测试组建的网路的性能，假设相邻的路由之间的数据传输需要一单位时间，现在需要知道任意两个路由之间传输数据最多需要多长时间。

输入格式

第一行一个整数 n(2≤n≤50000) 表示网络中路由个数。接下来 n - 1 行，每行输入 u,v(1≤u,v≤n) ，表示路由 u, v 相连。

输出格式

输出一行表示答案。

样例输入
8
6 3
3 7
3 4
7 5
5 1
6 8
5 2

样例输出
5

简要分析

开始没看懂，后来看了下网上大佬的解释，就明白怎么求树的直径（两次bfs）
下面简要描述下过程

1.第一次bfs任意以一个点为起始点,求出最远距离的顶点，这个顶点一定是树的直径的一个端点，也就是构成树的最远距离的两个点中的其中一个点。具体论证过程传送门

2.第二次bfs就用第一次求得的端点为起点，直接bfs求最远距离就可以啦。

ac代码

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
using namespace std;

const int MAXN=50001;
struct node{//点的结构体
    int num,step;//点的序号，存储步数
}point[MAXN];
struct edge{//边的结构体
    int v,next;
    edge(){}
    edge(int _v,int _next):v(_v),next(_next){}//构造函数
}e[MAXN*2];
int p[MAXN],maxp,maxs=0;;//存储最大距离的点数和步数
bool vst[MAXN];//判断是否拜访

void bfs(node a){
    //printf("2");
    queue<node> q;
    q.push(a);//将第一个元素放入
    //printf("a.num=%d a.step=%d\n",a.num,a.step);
    vst[a.num]=true;//设置为不可再拜访
    while(!q.empty()){
        //printf("1");
        struct node u=q.front();//取队首
        //printf("%d %d....",u.num,u.step);
        q.pop();//弹出队首
        for(int i=p[u.num];i+1;i=e[i].next){//邻接表的常规用法
            //printf("v=%d\n",e[i].v);
            if(!vst[e[i].v]){
                //printf("visit %d\n",e[i].v);
                vst[e[i].v]=true;//设置为不可再拜访
                point[e[i].v].step=u.step+1;
                point[e[i].v].num=e[i].v;
                if(maxs<point[e[i].v].step){
                    maxp=e[i].v;
                    maxs=point[e[i].v].step;
                }
                q.push(point[e[i].v]);
            }
        }
    }
}

int main()
{
    int n,u,v;
    scanf("%d",&n);
    memset(p,-1,sizeof(p));
    for(int i=0;i<2*(n-1);){//用邻接表存储边
        scanf("%d%d",&u,&v);
        e[i]=edge(v,p[u]);
        p[u]=i++;
        e[i]=edge(u,p[v]);
        p[v]=i++;
    }
    memset(point,0,sizeof(point));
    memset(vst,false,sizeof(vst));
    point[1].num=1;
    bfs(point[1]);
    //printf("%d %d\n",maxp,maxs);
    memset(point,0,sizeof(point));
    memset(vst,false,sizeof(vst));
    bfs(point[maxp]);
    printf("%d\n",maxs);
}