思路
从每个节点依次构树,选最大者就好了(height数组记录每个节点所构树的高度),如果在构树的过程中有节点没有访问到,说明不连通,此时在没有访问的节点处再次dfs构树,依次下去即可得到连通分支个数。
#include <cstdio>
#include <vector>
#include <algorithm>
using namespace std;
int N, visited[10010] = {0}, height[10010];
vector<int> G[10010];
int dfs(int i){
int h = 0;
visited[i] = 1;
for(int j = 0; j < G[i].size(); j++){
if(visited[G[i][j]] == 0){
//将高度修改为自己后代的最高者
h = max(dfs(G[i][j]), h);
}
}
return h + 1;
}
void getAns(){
int cnt = 1, maxheight = 0;
//分别以i为根构树
for(int i = 1; i <= N; i++){
memset(visited, 0, sizeof(visited));
height[i] = dfs(i);
maxheight = max(maxheight, height[i]);
//如果还有节点没有被访问到,说明图不连通
for(int j = 1; j <= N; j++){
if(visited[j] == 0){
//连通分支数加1
cnt++;
dfs(j);//构建连通分支
}
}
//cnt分支数大于1可以直接输出并结束程序
if(cnt > 1){
printf("Error: %d components\n", cnt);
return;
}
}
for(int i = 1; i <= N; i++)
if(height[i] == maxheight)
printf("%d\n", i);
}
int main(){
scanf("%d", &N);
for(int i = 1; i < N; i++){
int a, b;
scanf("%d%d", &a, &b);
G[a].push_back(b);
G[b].push_back(a);
}
getAns();
return 0;
}