1498. 最深的根

#include<bits/stdc++.h>
using namespace std;
const int N=1e4+10;
int h[N],e[N*2],ne[N*2],idx;
void add(int a,int b){
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
int n;
int p[N];
int find(int x){
    if(p[x]!=x)p[x]=find(p[x]);
    return p[x];
}
int dfs(int u,int father){
    int depth=0;
    for(int i=h[u];~i;i=ne[i]){
        int j=e[i];
        if(father!=j){
            int t=dfs(j,u);
            depth=max(t,depth);
        }
    }
    return depth+1;
}
int main(){
cin>>n;
int m=n-1;
memset(h,-1,sizeof h);
    for(int i=1;i<=n;i++)
    p[i]=i;
while(m--){
    int a,b;

    scanf("%d%d",&a,&b);
    add(a,b),add(b,a);
    if(find(a)!=find(b)){
        p[find(a)]=find(b);
    }
}   
    int cnt=0;
    for(int i=1;i<=n;i++){
        if(p[i]==i)cnt++;
    }   
    if(cnt>1)printf("Error: %d components",cnt);
    else{
        int maxdep=-1;
        vector<int>alls;
        for(int i=1;i<=n;i++){
            int t=dfs(i,-1);
            if(t>maxdep){
                alls.clear();
                alls.push_back(i);
                maxdep=t;
            }
            else if(t==maxdep)alls.push_back(i);
        }
        for(auto &item:alls)printf("%d\n",item);
    }
    return 0;
}