树形DP--------树的中心

给定一棵树，树中包含 nn 个结点（编号11~nn）和 n−1n−1 条无向边，每条边都有一个权值。
请你在树中找到一个点，使得该点到树中其他结点的最远距离最近。
输入格式
第一行包含整数 nn。
接下来 n−1n−1 行，每行包含三个整数 ai,bi,ciai,bi,ci，表示点 aiai 和 bibi 之间存在一条权值为 cici 的边。
输出格式
输出一个整数，表示所求点到树中其他结点的最远距离。
数据范围
1≤n≤100001≤n≤10000,

1≤ai,bi≤n1≤ai,bi≤n,

1≤ci≤1051≤ci≤105
输入样例：
5
2 1 1
3 2 1
4 3 1
5 1 1

输出样例：
2

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 10010, M = 2 * N,  INF = 0x3f3f3f3f;
int h[N], e[M], ne[M], w[M], idx;
int d1[N], d2[N], p1[N], up[N];
bool is_leap[N];
void add(int a, int b, int c){
 e[idx] = b, w[idx] = c, ne[idx] = h[a], h[a] = idx ++;
}
int dfs_d(int u, int father){
 d1[u] = d2[u] = -INF;
 for (int i = h[u]; ~i; i = ne[i]){
  int j = e[i];
   if (j == father)   continue;
  int d = dfs_d(j, u) + w[i];
  if (d > d1[u]){
   d2[u] = d1[u], d1[u] = d;
   p1[u] = j;
   }
   else  if (d > d2[u])    d2[u] = d;
 }
 if (d1[u] == -INF){
  d1[u] = d2[u] = 0;
  is_leap[u] = true;
 }
 return d1[u];
}
void dfs_u(int u, int father){
 for (int i = h[u]; ~i; i = ne[i]){
  int j = e[i];
  if (j == father)   continue;
  if (p1[u] == j){
   up[j] = max(up[u], d2[u]) + w[i];
  }
  else    up[j] = max(up[u], d1[u]) + w[i];
  dfs_u(j, u);
 }
}
int main(){
 int n;
 scanf("%d", &n);
 memset(h, -1, sizeof h);
 for (int i = 0; i < n - 1; i ++){
  int a, b, c;
  scanf("%d%d%d", &a, &b, &c);
  add(a, b, c), add(b, a, c);
 }
  dfs_d(1, -1);
 dfs_u(1, -1);
  int ans = d1[1];
 for (int i = 2; i <= n; i ++)
 if (is_leap[i])   ans = min(ans, up[i]);
 else              ans = min(ans, max(up[i], d1[i]));
  cout << ans << endl;
  return 0;
}