PAT甲级真题1021 最深的根（网上题解最好懂）

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

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

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

它被称为最深的根。

输入格式
第一行包含整数 NN，表示节点数量。

节点编号为 1∼N1∼N。

接下来 N−1N−1 行，每行包含两个整数，表示两个节点之间存在一条边。

输出格式
输出最深的根的节点编号。

如果最深的根不唯一，则按照从小到大的顺序，将它们依次输出，每个占一行。

如果给定的图不是树，输出 Error: K components，其中 KK 是图中连通分量的数量。

数据范围
1≤N≤1041≤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<bits/stdc++.h>
using namespace std;
const int N=10010;
int p[N];
vector<int>v[N];
set<int>s;
int find(int x){
	return(x==p[x])? x:p[x]=find(p[x]);
}
int dfs(int t,int father){	
	int dep=0;
	for(int i=0;i<v[t].size();i++){
		int e=v[t][i];
		if(father==e)continue;
		dep=max(dep,dfs(e,t)+1);//!这里取max
	}
	return dep;
}
int main(){
	int n;
	cin>>n;
	int cnt=n,maxd=-1; //连通块数量 
	for(int i=1;i<=n;i++)p[i]=i;//初始化
	for(int i=0;i<n-1;i++){
		int a,b;
		cin>>a>>b;
		v[a].push_back(b);
		v[b].push_back(a);
		if(find(a)!=find(b)){
		p[find(a)]=p[find(b)];
		cnt--;
	  }
	} 
	if(cnt>1){ //连通块大于一不是树
		printf("Error: %d components",cnt);
	}
	else{
	for(int i=1;i<=n;i++){  //每个点作为根节点 
		int deep=dfs(i,-1); //根的父节点是-1
		if(deep>maxd){
			s.clear();
			maxd=deep;
			s.insert(i);
		}
		else if(deep==maxd){
			s.insert(i);
		}	
	}
}
	for(auto it:s){
		cout<<it<<endl;
	}
	return 0;	
}