python/TangDou/AcWing/Tree/1073_2.cpp

#include <bits/stdc++.h>
using namespace std;

const int N = 10010;
const int M = N << 1;
const int INF = 0x3f3f3f3f;

int n;

int d1[N]; // d1[u]：存下u节点向下走的最长路径的长度
int d2[N]; // d2[u]：存下u节点向下走的次长路径的长度
int p1[N]; // p1[u]：存下u节点向下走的最长路径是从哪一个节点下去的
int up[N]; // up[u]：存下u节点向上走的最长路径的长度
int st[N];
// 邻接表
int e[M], h[N], idx, w[M], ne[M];
void add(int a, int b, int c = 0) {
    e[idx] = b, ne[idx] = h[a], w[idx] = c, h[a] = idx++;
}

// 功能:以u为根，向叶子进行递归，利用子节点返回的最长信息，更新自己的最长和次长,并记录最长是从哪个节点来的
void dfs1(int u) {
    st[u] = 1;
    for (int i = h[u]; ~i; i = ne[i]) {
        int j = e[i];
        if (st[j]) continue;

        // 递归完才能有数据
        dfs1(j);

        if (d1[j] + w[i] >= d1[u]) {     // 更新最长
            d2[u] = d1[u];               // ① 更新次长,必须在第一位，因为下面d1[u]会被改写
            d1[u] = d1[j] + w[i];        // ② 更新最长
            p1[u] = j;                   // ③ 记录最长来源
        } else if (d1[j] + w[i] > d2[u]) // 更新次长
            d2[u] = d1[j] + w[i];
    }
}

// 功能：完成向上的信息填充
void dfs2(int u) {
    st[u] = 1;
    for (int i = h[u]; ~i; i = ne[i]) {
        int j = e[i];
        if (st[j]) continue;

        // 三者取其一
        up[j] = w[i] + up[u];
        if (p1[u] == j)
            up[j] = max(up[j], w[i] + d2[u]);
        else
            up[j] = max(up[j], w[i] + d1[u]);
        // 准备好了信息，再进入递归
        dfs2(j);
    }
}

int main() {
    memset(h, -1, sizeof h);
    cin >> n;
    for (int i = 1; i < n; i++) {
        int a, b, c;
        cin >> a >> b >> c;
        add(a, b, c), add(b, a, c);
    }
    memset(st, 0, sizeof st);
    dfs1(1); // 选择任意一个节点进行dfs,用儿子更新父亲的统计信息
    memset(st, 0, sizeof st);
    dfs2(1); // 向上

    int res = INF;
    for (int i = 1; i <= n; i++) res = min(res, max(d1[i], up[i]));
    printf("%d\n", res);

    return 0;
}