树的直径模板

最新推荐文章于 2025-09-21 17:51:22 发布

原创最新推荐文章于 2025-09-21 17:51:22 发布 · 949 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#树的直径 #NOIPRP++

|NOIPRP++| 同时被 3 个专栏收录

65 篇文章

订阅专栏

>算法集合(_▽_)<

34 篇文章

订阅专栏

搜索

4 篇文章

订阅专栏

本文介绍了一种求解树上最长链的经典算法，并提供了完整的C++实现代码。通过两次深度优先搜索（DFS），首先找到距离根节点最远的点，再从此点出发寻找树的直径。

树的直径模板

http://codevs.cn/problem/1814/最长链.模板题
对于树上的最长链，有且仅有两种情况：
这里写图片描述

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
int n,tot,a,b,ans,pos;
int first[200010],nxt[200010];
struct edge
{
    int u,v;
}l[200010];
void build(int f,int t)
{
    l[++tot]=(edge){f,t};
    nxt[tot]=first[f];
    first[f]=tot;
}
void dfs(int k,int fa,int d)
{
    if(d>ans)
    {
        ans=d;
        pos=k;
    }
    for(int i=first[k];i!=-1;i=nxt[i])
    {
        int x=l[i].v;
        if(x==fa)
        continue;
        dfs(x,k,d+1);
    }
}
int main()
{
    memset(first,-1,sizeof(first));
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d%d",&a,&b);
        if(a!=0)
        {
            build(i,a);
            build(a,i);
        }
        if(b!=0)
        {
            build(i,b);
            build(b,i);
        }
    }
    dfs(1,1,0);//第一遍DFS找出树上深度最深的点
    ans=0;
    dfs(pos,1,0);//第二遍DFS由深度最深的点向上找最长链
    printf("%d",ans);
    return 0;   
}