AcWing 1498. 最深的根
原题链接
中等
作者:
leo123456
,
2020-08-21 17:18:46
,
所有人可见
,
阅读 564
#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int N=10010;
int n;
vector<int> son[N];
vector<int> res;
bool st[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 depth=0;
// st[u]=true;
//
// for(int i=0;i<son[u].size();i++)
// {
// if(st[son[u][i]]==true) continue;
// int h=dfs(son([u][i]));
// depth=max(depth,h);
// }
// st[u]=false;
// return depth+1;
//}
int dfs(int u,int father)
{
int depth=0;
for(int i=0;i<son[u].size();i++) u到叶子结点的最大高度=u的每个子节点到叶子结点的最大高度取一个max+1
{
int t=son[u][i];
if(t==father) continue; //走回头路了,就直接跳过
depth=max(depth,dfs(t,u)+1);
}
return depth;
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++) p[i]=i;
int k=n;
for(int i=0;i<n-1;i++)
{
int a,b;
cin>>a>>b;
son[a].push_back(b);
son[b].push_back(a);
if(find(a)!=find(b))
{
k--;
p[find(a)]=find(b);
}
}
if(k>1) printf("Error: %d components", k);
else
{
int max_depth=0;
for(int i=1;i<=n;i++)
{
int depth=dfs(i,-1);
if(depth>max_depth)
{
max_depth=depth;
res.clear();
res.push_back(i);
}
else if(depth==max_depth)
res.push_back(i);
}
for(auto v:res)cout<<v<<endl;
}
return 0;
}