树形DP求树的直径 【模板】

 

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
int const MAX = 100005;
int head[MAX], dp[MAX][2];
int n, s, cnt, ans;

struct EDGE
{
    int v, w, next;
}e[MAX];

void Add(int u, int v, int w)
{
    e[cnt].v = v;
    e[cnt].w = w;
    e[cnt].next = head[u];
    head[u] = cnt ++;
}

void DFS(int u, int fa)
{
    dp[u][0] = dp[u][1] = 0;
    for(int i = head[u]; i != -1; i = e[i].next)
    {
        int v = e[i].v;
        int w = e[i].w;
        if(v != fa)
        {
            DFS(v, u);
            if(dp[u][0] < dp[v][0] + w)
            {
                int tmp = dp[u][0];
                dp[u][0] = dp[v][0] + w;
                dp[u][1] = tmp;
            }
            else if(dp[u][1] < dp[v][0] + w)
                dp[u][1] = dp[v][0] + w;
        }
    }
    ans = max(ans, dp[u][1] + dp[u][0]);
    return;
}

int main()
{
    cnt = 0;
    ans = 0;
    memset(head, -1, sizeof(head));
    scanf("%d", &n);
    for(int i = 0; i < n - 1; i++)
    {
        int u, v, w;
        scanf("%d %d %d", &u, &v, &w);
        Add(u, v, w);
        Add(v, u, w);
    }
    DFS(1, -1);
    printf("%d\n",ans);
}