树的最长路径(树上dp，dfs)边权

传送门

题意：

给定一棵树，树中包含 n 个结点（编号1~n）和 n−1 条无向边，每条边都有一个权值。
现在请你找到树中的一条最长路径。
换句话说，要找到一条路径，使得使得路径两端的点的距离最远。
注意：路径中可以只包含一个点。

代码：

dp[x],表示从x出发的子树中的最长路径

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <string.h>
#include <vector>
#include <math.h>
#include <map>
#include <queue>
#include <set>
#include <stack>
typedef long long ll;
#define Pll make_pair
#define PB push_back
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define per(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int MAXN=1e5+50;
ll ans;
int f[MAXN],to[MAXN],nex[MAXN],val[MAXN];
int cnt;
ll dp[MAXN];
void add(int a,int b,int c){
    to[++cnt]=b;
    val[cnt]=c;
    nex[cnt]=f[a];
    f[a]=cnt;
}
void dfs(int x,int fa){
    ll ma=0;
    for(int i=f[x];i;i=nex[i]){
        int y=to[i];
        if(y==fa)continue;
        dfs(y,x);
        //ans=max(ans,ma+dp[y]+val[i]);
        //ma=max(ma,dp[y]+val[i]);
        ans=max(ans,dp[x]+dp[y]+val[i]);//经过x的加上经过y的，还有x,y之间的边
        dp[x]=max(dp[x],dp[y]+val[i]);
    }
    //dp[x]=ma;
}
int main()
{
    int n;
    scanf("%d",&n);
    rep(i,1,n-1){
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);
        add(b,a,c);
    }
    ans=0;
    dfs(1,-1);
    printf("%lld\n",ans);
    return 0;
}