【树形dp】树的中心

给定一棵树，树中包含 nn 个结点（编号1~n）和 n−1 条无向边，每条边都有一个权值。

请你在树中找到一个点，使得该点到树中其他结点的最远距离最近。

输入格式

第一行包含整数 n。

接下来 n−1行，每行包含三个整数 ai,bi,ci表示点 ai和 bi之间存在一条权值为 ci 的边。

输出格式

输出一个整数，表示所求点到树中其他结点的最远距离。

数据范围

1≤n≤10000
1≤ai,bi≤n
1≤ci≤1e5

输入样例：

5 
2 1 1 
3 2 1 
4 3 1 
5 1 1

输出样例：

2

 

 

思路：随便拿出一个点作为根节点，先用普通的树形dp用子节点更新父节点，求出每个点的子结点中的最长链与次长链。然后用父节点更新子节点，求出每个点往上走的最长链。因为这个点可能是父节点最长链的取值点，所以还需要记录父节点的最长链取自哪个子节点。详见代码及注释

 

#include<bits/stdc++.h>
#define ll long long
#define inf 1e9+7
using namespace std;
const int maxn = 1e4+10;
struct dd{
    int to,val;
};
vector<dd>q[maxn];
int n,a,b,c;
int d1[maxn],d2[maxn],up[maxn];
//d1用于记录每个点往下走的最长链，d2用于记录每个点往下走的次长链
//up用于记录每个点往上走的最长链
int book[maxn],p[m