1072. 树的最长路径 - AcWing题库

什么叫树的直径？

就是在一个树中求一个点到另一个点经过多少条边，这个最大的路径就是树的直径。

树的直径的解法

一般有两种解法：

①两遍dfs/bfs

②一遍dfs/bfs

code

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int h[N],e[N],ne[N],w[N],idx;
int n,ans;
bool st[N];

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

int dfs(int u)
{
    int dist=0;
    int d1=0,d2=0;  //最大值，次大值
    st[u]=true;
    
    for(int i=h[u];i!=-1;i=ne[i])
    {
        int j=e[i];
        
        if(!st[j])  //如果j这条边没有被走过的话
        {
            int d=dfs(j)+w[i];  //走一下
            dist=max(dist,d);   
            
            if(d>=d1) d2=d1,d1=d;
            else if(d>d2) d2=d;
        }
    }
    
    ans=max(ans,d1+d2);
    return dist;
}

int main()
{
    memset(h,-1,sizeof h);  //初始化头节点
    
    cin>>n;
    for(int i=0;i<n-1;i++)
    {
        int a,b,c;cin>>a>>b>>c;
        add(a,b,c),add(b,a,c);
    }
    
    dfs(1);
    
    
    cout<<ans<<endl;
    
    return 0;
}