#10159. 「一本通 5.2 练习 2」旅游规划

思路：判断该点是否可以构成树的直径 ，如果可以则该点位于城市交通网的最长路径上

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+5;
int d1[maxn],d2[maxn],p[maxn],up[maxn];
int n,num,mx;
int nx[maxn],head[maxn],v[maxn];
void add(int a,int b)
{
	v[num]=b;
	nx[num]=head[a];
	head[a]=num++;
}
void dfs1(int u,int fa)
{
	d1[u]=0;//d1[u]记录从u点向下走的最大长度 
	d2[u]=0;//d2[u]记录从u点向下走的次大长度 
	for(int i=head[u];i!=-1;i=nx[i])
	{
		int h=v[i];
		if(h==fa)//避免向上查找 
		continue;
		dfs1(h,u);
		//p[u]记录从u点向下走的最长路径是从那个点下去的 
		if(d1[h]+1>d1[u])
		d2[u]=d1[u],d1[u]=d1[h]+1,p[u]=h;
		else if(d1[h]+1>d2[u])
		d2[u]=d1[h]+1;
	}
	mx=max(d1[u]+d2[u],mx);
}
void dfs2(int u,int fa)
{
	
	for(int i=head[u];i!=-1;i=nx[i])
	{
		int h=v[i];//up[h]记录从h点向上走的最大长度 
		if(h==fa)
		continue;
		up[h]=up[u]+1;
		if(p[u]==h)
		up[h]=max(up[h],d2[u]+1);
		else
		up[h]=max(up[h],d1[u]+1);
		dfs2(h,u);
	}
}
int main()
{
	scanf("%d",&n);
	memset(head,-1,sizeof(head));
	for(int i=1;i<n;i++)
	{
	int x,y;
	scanf("%d%d",&x,&y);
	add(x,y);
	add(y,x);	
	}	
	dfs1(0,-1);
	dfs2(0,-1);
	for(int i=0;i<n;i++)
	{
		int e[3]={d1[i],d2[i],up[i]};
		sort(e,e+3);
		if(e[1]+e[2]==mx)//判断是否可以构成树的直径 
		printf("%d\n",i);
	}
}