有一点点繁杂,这个题。
我用的是遍历每个节点,用maxn记录最大数,如果更新,则clear输出数组。
这里有两次最大值判断,一个是每次dfs中的最长的路径,一个是每次dfs之后,把得到的结果最大值找出来,记录这次的root。
柳神的办法是,用两次dfs即可,分别找到最深结点,和最深结点依次遍历出最深结点。
#include<bits/stdc++.h>
using namespace std;
vector<int> t[10010];
bool flag[10010];
int maxn;
void dfs(int root,int d){
flag[root]=true;
if(d>maxn){
maxn=d;
}
for(int i=0;i<t[root].size();i++){
if(flag[t[root][i]]==false)dfs(t[root][i],d+1);
}
}
int main(){
int n,a,b;
cin>>n;
for(int i=0;i<n-1;i++){
cin>>a>>b;
t[a].push_back(b);
t[b].push_back(a);
}
vector<int> ans;
maxn=0;
int cnt=0;
for(int i=1;i<=n;i++){
if(flag[i]==false){
dfs(i,1);
cnt++;
}
}
if(cnt>1){
printf("Error: %d components",cnt);
return 0;
}
for(int i=1;i<=n;i++){
fill(flag,flag+10010,false);
int tag=maxn;
maxn=0;
dfs(i,1);
if(tag==maxn){
ans.push_back(i);
}
else if(tag<maxn){
ans.clear();
ans.push_back(i);
}
else{
maxn=tag;
}
}
for(int i=0;i<ans.size();i++){
printf("%d\n",ans[i]);
}
return 0;
}