PAT-A 1021 Deepest Root (25 分)

一个无环连通图可以被视作一个树。

树的高度取决于所选取的根节点。

现在，你要找到可以使得树的高度最大的根节点。

它被称为最深的根。

输入格式
第一行包含整数 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

思路：（1）判断一下连通分量（2）从1号节点开始找到最深的节点（3）然后从找到的其中之一继续寻找最深的节点（4）将2、3步的找到的最深的节点合并到一个set中输出。

#include <bits/stdc++.h>

using namespace std;
const int N = 10010;
int vis[N];
vector<int> v[N];
set<int> s;
vector<int> ans;
int maxx=0;
int n;
void dfs(int x)
{
    vis[x]=1;
    for(int each:v[x])
    {
        if(vis[each]==0)
        dfs(each);
    }
}
void dfs2(int id,int level)
{
  
    if(level>maxx)
    {
        ans.clear();
        ans.push_back(id);
        maxx=level;
    }
    else if(level==maxx)
    {
        ans.push_back(id);
    }
    for(int each:v[id])
    {
        if(vis[each]==0)
        {
            vis[each]=1;
            dfs2(each,level+1);
            vis[each]=0;
        }
    }
    
}
int main()
{
    cin >> n;
    for(int i=1;i<n;i++)
    {
        int x,y;
        cin >> x >>y;
        v[x].push_back(y);
        v[y].push_back(x);
    }
    int cnt=0;
    for(int i=1;i<=n;i++)
    {
        if(vis[i]==0)
        {  
            cnt++;
            dfs(i);
        }
    }
    if(cnt==1)
    {
        memset(vis,0,sizeof(vis));
        vis[1]=1;
        dfs2(1,0);
        for(int each:ans)
        {
            s.insert(each);
        }
        ans.clear();
          memset(vis,0,sizeof(vis));
        vis[*s.begin()]=1;
        maxx=0;
        dfs2(*s.begin(),0);
         for(int each:ans)
        {
            s.insert(each);
        }
        for(int it:s)
        cout <<it<<endl;
    }
    else if(cnt>1)
    {
       printf("Error: %d components\n",cnt);
    }
    return 0;
}