LuoguP1131 [ZJOI2007]时态同步

注意开longlong

参考神仙题解

自己的风格：

#include<cstdio>
#include<cstring>
#include<algorithm>
#define int long long
using namespace std;

int n,s,head[1000005],cnt,ans,mdtl[1000005];//max_dis_to_leaves

struct edge{
    int v,w,next;
}e[1000005];

inline void add(int u,int v,int w){
    e[++cnt].v=v;
    e[cnt].w=w;
    e[cnt].next=head[u];
    head[u]=cnt;
}

inline void dfs(int u,int fa){
    for(int i=head[u];i!=-1;i=e[i].next){
        int v=e[i].v;
        if(v==fa)continue;
        dfs(v,u);
        mdtl[u]=max(mdtl[u],mdtl[v]+e[i].w);
    }
    for(int i=head[u];i!=-1;i=e[i].next){
        int v=e[i].v;
        if(v==fa)continue;
        ans+=mdtl[u]-(mdtl[v]+e[i].w);
    }
}

signed main(){
    memset(head,-1,sizeof(head));
    scanf("%lld%lld",&n,&s);
    for(int i=1;i<=n-1;i++){
        int a,b,t;
        scanf("%lld%lld%lld",&a,&b,&t);
        add(a,b,t);
        add(b,a,t);
    }
    dfs(s,0);
    printf("%lld\n",ans);
}

转载于:https://www.cnblogs.com/Y15BeTa/p/11294117.html