樹的最長路徑(樹上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;
}