树的直径

参考：树的直径

树的直径：树上最远两点（叶子结点）的距离

算法：从树上任意点u开始DFS(BFS)遍历图，得到距离u最远的结点v,然后从v点开始DFS遍历图，得到距离v最远的结点w， 则v、w之间的距离就是树的直径，且复杂度为2趟DFS，因此复杂度为 O(n)

重要性质：树上任意点能到的最远点，一定是树的直径的某个端点。

模板：

 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e5+10;
const int inf = 0x3f3f3f3f;
const int mod = 998244353;
#define rep(i,first,second) for(int i=first;i<=second;i++)
#define dep(i,first,second) for(int i=first;i>=second;i--)

struct edge{ int to,nxt,val;}e[maxn<<1];
int head[maxn<<1],maxdis,maxv,tot;
int n;

void add(int u,int v,int w){
    e[tot].to = v;
    e[tot].nxt = head[u];
    e[tot].val = w;
    head[u] = tot++;
}

//u:dfs到的点; f:u点的父节点; dis:u到源点的距离
void dfs(int u,int f,int dis){
    if( maxdis<dis ){
        maxdis = dis;
        maxv = u;
    }
    for(int i=head[u];~i;i=e[i].nxt){
        int v=e[i].to, w=e[i].val;
        if( v==f ) continue;//父节点已经访问过,防止重复遍历
        dfs(v,u,dis+w);
    }
}

int main()
{
    scanf("%d",&n);
    memset(head,-1,sizeof(head));
    rep(i,1,n){
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        add(u,v,w);
        add(v,u,w);
    }
    maxv=1;
    maxdis=0;
    dfs(1,-1,0);//从结点1开始遍历,找到离1最远的点maxv及对应的最远距离maxdis
    maxdis=0;
    dfs(maxv,-1,0);//从结点maxv开始遍历，找到最远点对应的距离maxdis
    printf("%d\n",maxdis);
    return 0;
}