1021. Deepest Root (25)
A graph which is connected and acyclic can be considered a tree. The height of the tree depends on the selected root. Now you are supposed to find the root that results in a highest tree. Such a root is called the deepest root.
Input Specification:
Each input file contains one test case. For each case, the first line contains a positive integer N (<=10000) which is the number of nodes, and hence the nodes are numbered from 1 to N. Then N-1 lines follow, each describes an edge by given the two adjacent nodes' numbers.
Output Specification:
For each test case, print each of the deepest roots in a line. If such a root is not unique, print them in increasing order of their numbers. In case that the given graph is not a tree, print "Error: K components" where K is the number of connected components in the graph.
Sample Input 1:5 1 2 1 3 1 4 2 5Sample Output 1:
3 4 5Sample Input 2:
5 1 3 1 4 2 5 3 4Sample Output 2:
Error: 2 components
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
vector<int> v[N];
int f[N];
bool vis[N];
int d[N];
int find(int x){
if(f[x]==-1)return x;
return f[x]=find(f[x]);
}
int dfs(int s){
int ans=0;
if(vis[s])return 0;
vis[s]=true;
for(auto u:v[s]){
if(!vis[u])
ans=max(ans,dfs(u));
}
return ans+1;
}
int main(){
int n;
scanf("%d",&n);
memset(f,-1,sizeof(f));
for(int i=1;i<n;i++){
int x,y;
scanf("%d%d",&x,&y);
v[x].push_back(y);
v[y].push_back(x);
x=find(x);
y=find(y);
if(x!=y)
f[y]=x;
}
int sum=0;
for(int i=1;i<=n;i++)
if(find(i)==i)sum++;
if(sum>1){
printf("Error: %d components\n",sum);return 0;
}
int mm=-1;
for(int i=1;i<=n;i++){
memset(vis,0,sizeof(vis));
d[i]=dfs(i);
mm=max(mm,d[i]);
}
bool f=false;
for(int i=1;i<=n;i++)
if(d[i]==mm){
printf("%d\n",i);
}
}