萌新求助

#include<cstdio>
#include<iostream>
using namespace std;
int n,s;
int cnt,st[500005];
long long ans,dis[500005];
struct edge{
    int v,val,last;
}e[1000001];
void add(int x,int y,int z){
    e[++cnt].v=y;
    e[cnt].val=z;
    e[cnt].last=st[x];
    st[x]=cnt;
}
void dfs(int u,int fa){
    for(int i=st[u];i;i=e[i].last){
        if(e[i].v==fa)
            continue;
        dfs(e[i].v,u);
        dis[u]=max(dis[u],dis[e[i].v]+e[cnt].val);
    }
    for(int i=st[u];i;i=e[i].last){
        if(e[i].v==fa)
            continue;
        ans+=dis[u]-dis[e[i].v]-e[i].val;
    }
}
int main(){
    scanf("%d %d",&n,&s);
    for(int i=1;i<n;i++){
        int a,b,t;
        scanf("%d %d %d",&a,&b,&t);
        add(a,b,t);add(b,a,t);
    }
    dfs(s,0);
    printf("%lld",ans);
    return 0;
}