bzoj 1060 [ZJOI2007]时态同步（树形DP）

【题目链接】

    http://www.lydsy.com/JudgeOnline/problem.php?id=1060

【题意】

    求最少的增加量，使得以rt为根的树中由一个结点出发的所有到叶子结点的路长相等。

【思路】

    树形DP。

    设f[u]为以u为根的子树中到叶子的最大路长，只要把其他的所有路长都增加到f[u]就可以了。

【代码】

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

typedef long long ll;
const int N = 5e5+10;

struct Edge {
    int v,w,nxt;
}e[N<<1];
int en=1,front[N];
void adde(int u,int v,int w) 
{
    en++; e[en].v=v,e[en].w=w,e[en].nxt=front[u],front[u]=en;
}

int n,rt;
int f[N]; ll ans;

ll read()
{
    char c=getchar(); ll f=1,x=0;
    while(!isdigit(c)) {if(c=='-')f=-1; c=getchar();}
    while(isdigit(c)) x=x*10+c-'0',c=getchar();
    return x*f;
}

void dfs(int u,int fa) {
    for(int i=front[u];i;i=e[i].nxt) {
        int v=e[i].v;
        if(v!=fa) {
            dfs(v,u);
            f[u]=max(f[u],f[v]+e[i].w);
        }
    }
    for(int i=front[u];i;i=e[i].nxt)
        if(e[i].v!=fa) ans+=f[u]-f[e[i].v]-e[i].w;
}

int main()
{
    n=read(),rt=read();
    for(int i=0;i<n-1;i++) {
        int u=read(),v=read(),w=read();
        adde(u,v,w),adde(v,u,w);
    }
    dfs(rt,-1);
    printf("%lld
",ans);
    return 0;
}

合影留念 :)