一个无环连通图可以被视作一个树。
树的高度取决于所选取的根节点。
现在,你要找到可以使得树的高度最大的根节点。
它被称为最深的根。
输入格式
第一行包含整数 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;
}