1498. 最深的根

一个无环连通图可以被视作一个树。树的高度取决于所选取的根节点。现在，你要找到可以使得树的高度最大的根节点。它被称为最深的根。

输入格式
第一行包含整数 N，表示节点数量。节点编号为 1∼N。接下来 N−1 行，每行包含两个整数，表示两个节点之间存在一条边。

输出格式
输出最深的根的节点编号。如果最深的根不唯一，则按照从小到大的顺序，将它们依次输出，每个占一行。如果给定的图不是树，输出 Error: K components，其中 K 是图中连通分量的数量。

数据范围
1≤N≤104

输入样例1：
5
1 2
1 3
1 4
2 5

输出样例1：
3
4
5

输入样例2：
5
1 3
1 4
2 5
3 4

输出样例2：
Error: 2 components

#include<iostream>
#include<vector>
#include<cstring>
using namespace std;
const int N=1e4+10,M=2*N;
int h[N],e[M],ne[M],fa[N];
int n,idx;
void add(int a,int b)
{
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
int find(int x)
{
    if(x!=fa[x]) fa[x]=find(fa[x]);
    return fa[x];
}
int dfs(int u,int fa)
{
    int height=0;
    for(int i=h[u];~i;i=ne[i])
    {
        int j=e[i];
        if(j==fa) continue;
        height=max(height,dfs(j,u)+1);
    }
    return height;
}
int main()
{
    cin>>n;
    memset(h,-1,sizeof h);
    for(int i=1;i<=n;i++) fa[i]=i;
    int cnt=n;
    for(int i=0;i<n-1;i++)
    {
        int a,b;
        cin>>a>>b;
        add(a,b),add(b,a);
        if(find(a)!=find(b))
        {
            cnt--;
            fa[find(a)]=find(b);
        }
    }
    if(cnt>1) printf("Error: %d components",cnt);
    else
    {
        int max_h=-1;
        vector<int> num;
        for(int i=1;i<=n;i++)
        {
            int high=dfs(i,-1);
            if(high>max_h)
            {
                max_h=high;
                num.clear();
                num.push_back(i);
            }
            else if(high==max_h)
                num.push_back(i);
        }
        for(auto v:num) cout<<v<<endl;
    }
    return 0;
}