hdu2196 (树的直径)_hdu 2196 直径-优快云博客

本文链接：https://blog.youkuaiyun.com/tree__water/article/details/52887810

本文介绍了一种通过三次深度优先搜索（DFS）来确定树中每个节点能够到达的最远距离的方法，并计算出树的直径。具体步骤包括寻找树中最远的端点之一，接着找到另一个端点，最后计算所有节点到这两个端点的距离。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

求树上每个点能到达的最远距离，刚开始，就想找到树上最远的两个个点，那么树上的每一个点能到达的最远距离就是到这两个点的距离的大的那一个，（后来发现树上最远的两个个点的距离就叫树的直径）。
写了三个dfs，先求出树上最远的距离，肯定是直径的一个端点，然后再以这端点为根，再跑一次dfs，求出另一个端点，顺便记录到每个点的距离，找出另一个端点，为根跑一遍dfs，记录出另一个端点到每个点的距离，取较大值即可。
不过貌似还有更好的办法，

我的写法：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=10005;
int s,t,ma,tmp,n;
struct node
{
    int nxt,to,w;
}edge[N<<1];
int head[N],cnt;
int ss[N],tt[N];
void add(int u,int v,int w)
{
    edge[++cnt].to=v;
    edge[cnt].w=w;
    edge[cnt].nxt=head[u];
    head[u]=cnt;
}
void dfs(int u,int pre,int d)
{
    if(d>ma)
    {
        ma=d;
        tmp=u;
    }
    for(int i=head[u];i;i=edge[i].nxt)
    {
        int v=edge[i].to;
        if(v==pre) continue;
        dfs(v,u,d+edge[i].w);
    }
}
void dfs_1(int u,int pre,int d)
{
    if(d>ma)
    {
        ma=d;
        tmp=u;
    }
    ss[u]=d;
    for(int i=head[u];i;i=edge[i].nxt)
    {
        int v=edge[i].to;
        if(v==pre) continue;
        dfs_1(v,u,d+edge[i].w);
    }
}
void dfs_2(int u,int pre,int d)
{
    tt[u]=d;
    for(int i=head[u];i;i=edge[i].nxt)
    {
        int v=edge[i].to;
        if(v==pre) continue;
        dfs_2(v,u,d+edge[i].w);
    }
}
int main()
{
    while(~scanf("%d",&n))
    {
        cnt=0;
        memset(head,0,sizeof(head));
        int u,w;
        for(int i=2;i<=n;i++)
            scanf("%d%d",&u,&w),add(i,u,w),add(u,i,w);
        ma=-1;
        dfs(1,0,0);
        s=tmp;ma=-1;
        dfs_1(s,0,0);
        t=tmp;
        dfs_2(t,0,0);

        for(int i=1;i<=n;i++)
            printf("%d\n",max(ss[i],tt[i]));
    }
}