AcWing 1498. 最深的根
原题链接
中等
作者:
ls131
,
2021-04-10 19:19:48
,
所有人可见
,
阅读 1139
#include<bits/stdc++.h>
using namespace std;
const int N=1e4+10,M=N*2; //
int h[N],e[M],ne[M],idx; //邻接表存稀疏树图
int p[N]; //并查集 查找连通块
int n;
int find(int x)
{
if(p[x]!=x) p[x]=find(p[x]);
return p[x];
}
void add(int a,int b)
{
e[idx]=b;
ne[idx]=h[a];
h[a]=idx++;
}
int dfs(int u,int father) //记录来源点 因为无向图可能重复搜
{
int depth =0;
for(int i=h[u];~i;i=ne[i])
{
int j=e[i];
if(j==father) continue; //重复搜了
depth=max(depth,dfs(j,u)+1);
}
return depth;
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++) p[i]=i; //初始化并查集
memset(h,-1,sizeof h); //初始化邻接表
int k=n;
for(int i=0;i<n-1;i++)
{
int a,b;
cin>>a>>b;
if(find(a)!=find(b))
{
p[find(a)]=find(b); //并查集中的操作
k--;
}
add(a,b);add(b,a); //无向边加两次
}
if(k>1) printf("Error: %d components\n",k);
else
{
vector<int>nodes;
int max_depth=-1; //设置相反界外量
for(int i=1;i<=n;i++)
{
int depth=dfs(i,-1); //这个节点无来源点 得到这个点的深度
if(depth>max_depth)
{
max_depth=depth; //更新
nodes.clear(); //清空
nodes.push_back(i);
}
else if(depth==max_depth) nodes.push_back(i);
}
for(auto v:nodes) cout<<v<<endl;
}
return 0;
}