You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

76 lines
2.2 KiB

This file contains ambiguous Unicode characters!

This file contains ambiguous Unicode characters that may be confused with others in your current locale. If your use case is intentional and legitimate, you can safely ignore this warning. Use the Escape button to highlight these characters.

#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;
}