NOIP 冲刺 模板：树的直径

题目描述
树的直径：树上两点之间的最大距离。

给出一个树，让你求树的直径。

输入

一个数n表示节点数，以下（n-1）行每行两个数x，y表示x与y间有边。

输出
一个整数，树的直径。

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

样例输出
5

提示

40% n<=2000

100% n<=200000

#include <cstdio>#include <iostream>#include <queue>#include <cstring>#define N 200005using namespace std;int n,tot;int nex[N*2];int to[N*2];int head[N*2];inline int read(){    int x=0,f=1;    char ch=getchar();    while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=getchar();}    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}    return x*f;}bool vis[N*2];int dis[N*2];inline void add(int x,int y){    ++tot;    nex[tot]=head[x];    head[x]=tot;    to[tot]=y;}inline int spfa(int u){    queue<int> q;    memset(dis,0x7f7f7f,sizeof(dis));    memset(vis,0,sizeof(vis));    q.push(u);    dis[u]=0;    vis[u]=1;    while(!q.empty()){        int dmf=q.front();        q.pop();        for(int i=head[dmf];i;i=nex[i]){            if(dis[to[i]]>dis[dmf]+1){                dis[to[i]]=dis[dmf]+1;                if(!vis[to[i]]){                    vis[to[i]]=1;                    q.push(to[i]);                }            }        }        vis[dmf]=0;    }    int maxn=0;    int id;    for(int i=1;i<=n;i++){        if(i!=u){            if(dis[i]>maxn){                id=i;                maxn=dis[i];            }        }    }    return id;}int main(){    n=read();    for(int i=1;i<n;i++){        int x,y;        x=read();        y=read();        add(x,y);        add(y,x);    }    int ans=0;    int id=spfa(1);    ans+=dis[id];    int ans1=ans;    int id2=spfa(id);    if(id==1) ans=max(ans,dis[id2]);    else ans+=dis[id2];    printf("%d",ans-ans1);    return 0;}