bzoj千题计划163：bzoj1060: [ZJOI2007]时态同步

#include<cstdio>
#include<iostream>

using namespace std;

#define N 500001

int front[N],to[N<<1],nxt[N<<1],tot,val[N<<1];

long long dep[N],mx[N];
long long tim;

long long ans,sum[N];

void read(int &x)
{
    x=0; char c=getchar();
    while(!isdigit(c)) c=getchar();
    while(isdigit(c)) { x=x*10+c-‘0‘; c=getchar(); }
}

void add(int u,int v,int w)
{
    to[++tot]=v; nxt[tot]=front[u]; front[u]=tot; val[tot]=w;
    to[++tot]=u; nxt[tot]=front[v]; front[v]=tot; val[tot]=w;
}

void dfs(int x,int y)
{
    mx[x]=dep[x];
    tim=max(tim,dep[x]);
    for(int i=front[x];i;i=nxt[i])
        if(to[i]!=y)
        {
            dep[to[i]]=dep[x]+val[i];
            dfs(to[i],x);
            mx[x]=max(mx[x],mx[to[i]]);
        }
}

void dfs2(int x,int y)
{
    ans+=tim-mx[x]-sum[x];
    sum[x]+=tim-mx[x]-sum[x];
    for(int i=front[x];i;i=nxt[i])
        if(to[i]!=y)
        {
            sum[to[i]]=sum[x];
            dfs2(to[i],x);
        }
}

int main()
{
    int n,root;
    read(n);
    read(root);
    int u,v,w;
    for(int i=1;i<n;++i)
    {
        read(u); read(v); read(w);
        add(u,v,w);
    }
    dfs(root,0);
    dfs2(root,0);
    cout<<ans;
}