这道题目有点水啊,只要深搜到叶子节点,然后回溯,找出一个点到它的儿子节点的最大距离, a n s + = ans+= ans+=最大距离 − - −到每个儿子节点距离,最后记得把它父亲节点到它的距离加上之前所求的最大距离。
#include<bits/stdc++.h>
#define il inline
#define ll long long
#define max(a,b) ((a)>(b))?(a):(b)
using namespace std;
const int N=5e5+5;
#define getchar()(p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
char buf[1<<21],*p1=buf,*p2=buf;
il int read(){
int x=0,f=1;char c=getchar();
for(;!isdigit(c);c=getchar()) if(c=='-') f=-1;
for(;isdigit(c);c=getchar()) x=(x+(x<<2)<<1)+c-48;
return x*f;
}
int n,s,k=1,h[N];ll ans;
struct Edge{int to,nxt,val;}a[N<<1];
il void add(int x,int y,int z){a[++k].to=y,a[k].nxt=h[x],a[k].val=z,h[x]=k;}
il void dfs(int x,int fa){
int mx=0;
for(int i=h[x];i;i=a[i].nxt)
if(a[i].to!=fa) dfs(a[i].to,x),mx=max(mx,a[i].val);
for(int i=h[x];i;i=a[i].nxt)
if(a[i].to!=fa) ans+=mx-a[i].val;
else a[i^1].val+=mx;
}
int main(){
n=read(),s=read();
for(int i=1;i<n;++i){
int x=read(),y=read(),z=read();
add(x,y,z);add(y,x,z);
}dfs(s,0);
return !printf("%lld",ans);
}