题意
给出
n
个点
思路
网上很多的解法用并查集判断有几个联通块,然后以每个叶子节点进行DFS,这样的复杂度是
O(n2)
的
其实只要两遍DFS就好了,顺便把联通块个数也判掉
- 用DFS判断联通块,这个应该很好理解
- 第一次DFS求出最深的叶子节点
- 第二次DFS从这些最深的叶子节点中选一个再进行DFS,再求出一批最深的叶子节点
- 答案是两次DFS结果的并集(第一次求出最长路的一侧,第二次求出另一侧,因为所有不同的最长路必有一个公共交点,可以用反证法证明)
代码
#include <bits/stdc++.h>
#define mem(a,b) memset(a,b,sizeof(a))
#define rep(i,a,b) for(int i=a;i<b;i++)
#define debug(a) printf("a =: %d\n",a);
const int INF=0x3f3f3f3f;
const int maxn=1e4+50;
const int Mod=1e9+7;
const double PI=acos(-1);
typedef long long ll;
typedef unsigned int ui;
using namespace std;
vector<int> G[maxn];
int n;
bool vis[maxn];
int depth[maxn];
void dfs(int cur,int dep){
if (vis[cur]) return;
vis[cur]= true;
depth[cur]=dep;
for(auto i:G[cur]){
dfs(i,dep+1);
}
}
void getMax(int &ret,int &pos){
ret=-1,pos=-1;
for(int i=1;i<=n;i++) {
if (depth[i]>ret){
ret=depth[i];
pos=i;
}
}
}
void getNum(int x,set<int> &v){
for(int i=1;i<=n;i++) if (depth[i]==x) v.insert(i);
}
int main()
{
#ifndef ONLINE_JUDGE
// freopen("in.txt","r",stdin);
#endif
while(~scanf("%d",&n)){
for(int i=1;i<=n;i++) G[i].clear();
for(int i=1,u,v;i<n;i++){
scanf("%d %d",&u,&v);
G[u].push_back(v);
G[v].push_back(u);
}
mem(vis,0);
int pt=0;
for(int i=1;i<=n;i++){
if(!vis[i]) {
dfs(i,1);
pt++;
}
}
if (pt>1){
printf("Error: %d components\n",pt);
continue;
}
set<int> v;
int mx,pos;
getMax(mx,pos);
getNum(mx,v);
mem(vis,0);
dfs(pos,1);
getMax(mx,pos);
getNum(mx,v);
for(auto i:v) printf("%d\n",i);
}
return 0;
}
测试数据
9
1 3
3 4
4 5
2 3
2 7
2 6
7 8
6 9