题意:
给定一棵树,树中包含 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;
}