树的直径

树的直径

树上最长的简单路径即为树的直径。

求树的直径的方法就是在树上任选一点u，求距离点u最远的点v，再求距离点v最远的点s，点u到点s的距离即为树的直径。（即两遍DFS或BFS求最远点）

证明：

1. 如果u 是直径上的点，则v显然是直径的终点(因为如果v不是的话，则必定存在另一个点w使得u到w的距离更长，则于BFS找到了v矛盾)
   2) 如果u不是直径上的点，则u到v必然于树的直径相交(反证),那么交点到v 必然就是直径的后半段了
   所以v一定是直径的一个端点，所以从v进行BFS得到的一定是直径长度。

#include<cstdio>
#include<cstring>
#define N 4200
struct hehe{
    int next;
    int to;
}edge[N];
int num_edge,head[N],dis[N],n,a,b,y;
int add_edge(int from,int to){
    edge[++num_edge].next=head[from];
    edge[num_edge].to=to;
    head[from]=num_edge;
}
int dfs(int x){
    for(int i=head[x];i;i=edge[i].next)
        if(!dis[edge[i].to]){
            dis[edge[i].to]=dis[x]+1;
            dfs(edge[i].to);
        }
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<n;++i){
        scanf("%d%d",&a,&b);
        add_edge(a,b);
        add_edge(b,a);
    }
    dfs(1);
    for(int i=y=1;i<=n;i++)
        if(dis[i]>dis[y])
            y=i;
    memset(dis,0,sizeof(dis));
    dfs(y);
    for(int i=y=1;i<=n;i++)
        if(dis[i]>dis[y])
            y=i;
    printf("%d",dis[y]);
    return 0;
}