bzoj 1060: [ZJOI2007]时态同步 树形dp

题意

给出一棵有根树，边有边权。每次操作可以使得某条边的长度+1，问多少次操作后所有叶节点的深度相等。
n<=500000

分析

直接树形dp即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N=500005;

int n,root,cnt,last[N];
LL dep[N],mx[N],ans;
struct edge{int to,w,next;}e[N*2];

void addedge(int u,int v,int w)
{
    e[++cnt].to=v;e[cnt].w=w;e[cnt].next=last[u];last[u]=cnt;
    e[++cnt].to=u;e[cnt].w=w;e[cnt].next=last[v];last[v]=cnt;
}

void dfs(int x,int fa)
{
    mx[x]=dep[x];
    for (int i=last[x];i;i=e[i].next)
    {
        if (e[i].to==fa) continue;
        dep[e[i].to]=dep[x]+e[i].w;
        dfs(e[i].to,x);
        mx[x]=max(mx[x],mx[e[i].to]);
    }
    for (int i=last[x];i;i=e[i].next)
    {
        if (e[i].to==fa) continue;
        ans+=mx[x]-mx[e[i].to];
    }
}

int main()
{
    scanf("%d%d",&n,&root);
    for (int i=1;i<n;i++)
    {
        int x,y,z;
        scanf("%d%d%d",&x,&y,&z);
        addedge(x,y,z);
    }
    dfs(root,0);
    printf("%lld",ans);
    return 0;
}