dp（旅游规划）

总结：求树的直径时，只需dfs_d,定义两个d1,d2数组，分别存储每个节点向下走的最长路径的长度和次长路径的长度。

所有结点的最长路径和次长路径之和的最大值就是树的直径

如果还要求树的直径上的所有点，还要dfs_u,定义一个up数组，表示每个节点向上走能走的最远距离

p1数组表示每一个节点向下的最长路径所经过的直接子节点。

更新方式为：dfs_u，画个图一目了然

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>

using namespace std;

const int N=200020,M=2*N;

int h[N],e[M],ne[M],idx;

int n;
int d1[N],d2[N],p[N],up[N];
int maxd;

void add(int a,int b)
{
    e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}

void dfs_d(int u,int father)
{
    for(int i=h[u];i!=-1;i=ne[i])
    {
        int j=e[i];
        if(j!=father)
        {
            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 father)
{
    for(int i=h[u];i!=-1;i=ne[i])
    {
        int j=e[i];
        if(j!=father)
        {
            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()
{
    cin>>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(maxd==d[1]+d[2]) printf("%d\n",i);
        
    }
    return 0;
}