acwing1072. 树的最长路径(树形dp)

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

现在请你找到树中的一条最长路径。

换句话说，要找到一条路径，使得使得路径两端的点的距离最远。

注意：路径中可以只包含一个点。

输入格式
第一行包含整数 n。

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

输出格式
输出一个整数，表示树的最长路径的长度。

数据范围
1≤n≤10000,
1≤ai,bi≤n,
−105≤ci≤105

输入样例：
6
5 1 6
1 4 5
6 3 9
2 6 8
6 1 7
输出样例：
22

题解
树形dp

#include<bits/stdc++.h>
using namespace std;
const int N = 1e4 + 10;
const int M = 2 * N;
const int INF = 0x3f3f3f3f;
struct Edge{
    int v,next,w;
}edge[M];
int head[N],cnt;
int res = 0;
void add(int u,int v,int w){
    edge[cnt].v = v;
    edge[cnt].w = w;
    edge[cnt].next = head[u];
    head[u] = cnt ++;
}
int dfs(int u,int fa){
    int d1 = 0,d2 = 0;
    for(int i = head[u];~i;i = edge[i].next){
        int v = edge[i].v,w = edge[i].w;
        if(v == fa)continue;
        int t = dfs(v,u) + w;
        if(t >= d1)d2 = d1,d1 = t;
        else if(t > d2)d2 = t;
    }
    res = max(res,d1 + d2);
    return d1;
}
int main(){
    int n;
    memset(head,-1,sizeof head);
    cin>>n;
    int x,y,w;
    for(int i = 0;i < n - 1;i ++){
        cin>>x>>y>>w;
        add(x,y,w);
        add(y,x,w);
    }
    dfs(1,-1);
    cout<<res<<endl;
    return 0;
}