洛谷 1131 [ZJOI2007]时态同步——树形dp

题目：https://www.luogu.org/problemnew/show/P1131

因为越高，调节一个影响到的越多，所以底下只要把子树间的差异消除了就行了，与其他部分的差异由更高的边调节。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=5e5+;
int n,rt,hd[N],xnt,to[N<<],nxt[N<<],w[N<<];
ll ans,dis[N<<];
int rdn()
{
    int ret=,fx=;char ch=getchar();
    while(ch>''||ch<''){if(ch=='-')fx=; ch=getchar();}
    while(ch>=''&&ch<='') ret=(ret<<)+(ret<<)+ch-'',ch=getchar();
    return fx?ret:-ret;
}
void add(int x,int y,int z)
{
    to[++xnt]=y; nxt[xnt]=hd[x]; hd[x]=xnt; w[xnt]=z;
    to[++xnt]=x; nxt[xnt]=hd[y]; hd[y]=xnt; w[xnt]=z;
}
void dfs(int cr,int fa)
{
    for(int i=hd[cr],v;i;i=nxt[i])
        if((v=to[i])!=fa)
        {
            dfs(v,cr);
            dis[cr]=max(dis[cr],dis[v]+w[i]);
        }
    for(int i=hd[cr],v;i;i=nxt[i])
        if((v=to[i])!=fa)
            ans+=dis[cr]-(dis[v]+w[i]);
}
int main()
{
    n=rdn(); rt=rdn();
    for(int i=,u,v,z;i<n;i++)
    {
        u=rdn(); v=rdn(); z=rdn(); add(u,v,z);
    }
    dfs(rt,);
    printf("%lld\n",ans);
    return ;
}