旅游规划(c++实现)
【代码】旅游规划(c++实现)
·
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int N = 2e5+10,M=N*2;
int h[N],e[M],ne[M],idx;
int n;
int d1[N],d2[N],up[N],p[N];
int maxd;
void add(int a,int b){
e[idx] = b;
ne[idx] = h[a];
h[a] = idx;
idx++;
}
void dfs_d(int u,int fa){
for(int i=h[u];i!=-1;i=ne[i]){
int j = e[i];
if(j!=fa){
dfs_d(j,u);
int distance = d1[j]+1;
if(distance > d1[u]){
d2[u] = d1[u];
d1[u] = distance;
p[u] = j;
}
else if(distance>d2[u]){
d2[u] = distance;
}
}
}
maxd = max(maxd,d1[u]+d2[u]);
}
void dfs_u(int u,int fa){
for(int i=h[u];i!=-1;i=ne[i]){
int j = e[i];
if(j!=fa){
up[j] = up[u] + 1;
if(p[u]==j) up[j] = max(up[j],d2[u]+1);
else up[j] = max(up[j],d1[u]+1);
dfs_u(j,u);
}
}
}
int main(){
scanf("%d",&n);
memset(h,-1,sizeof h);
for(int i=0;i<n-1;i++){
int a,b;
scanf("%d%d",&a,&b);
add(a,b),add(b,a);
}
dfs_d(0,-1);
dfs_u(0,-1);
for(int i=0;i<n;i++){
int d[3] = {d1[i],d2[i],up[i]};
sort(d,d+3);
if(d[1] + d[2] == maxd) printf("%d\n",i);
}
}
更多推荐
已为社区贡献20条内容
所有评论(0)