树的最长路径（树形DP）

原题链接
枚举每个起点，找到距离这个点最远的点，每次更新最小值

初始的时候不确定图的拓扑结构，因此要建立双向边

#include <iostream>
#include <cstring>

constexpr int N = 10005, M = N << 1;

int h[N], e[M], ne[M], w[M], idx;
int res;

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

// 找到以u为根节点到底下的最长路径
int dfs(int u, int father) {
    int d1 = 0, d2 = 0; // d1,d2为从u往下走的最长路径和次长路径
    for (int i = h[u]; ~i; i = ne[i]) {
        int j = e[i];
        if (j == father) continue; // 不能网上走，以免死循环

        int d = dfs(j, u) + w[i];
        if (d > d1) d2 = d1, d1 = d;
        else if (d > d2) d2 = d;
    }
    res = std::max(res, d1 + d2);

    return d1;
}

int main() {
    memset(h, -1, sizeof h);
    int n;
    std::cin >> n;
    for (int i = 0; i < n - 1; i ++ ) {
        int a, b, c;
        std::cin >> a >> b >> c;
        add(a, b, c), add(b, a, c);
    }

    dfs(1, -1); // 任取一点作为根节点
    std::cout << res << '\n';

    return 0;
}