题目描述

传送门

题解

这也叫树形dp？不过打个dfs貌似好low呀，所以还是叫它树形dp叭。
维护size不就可以了？这题放NOI有点水的说。
不过Po的代码在BZ上无端RE，不过在vijos上跑过了= =

2017/2/17update
bz爆栈了
写手工栈吧
sb

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define LL long long
#define N 1000005

int n,x,y; LL z,ans;
int tot,point[N],nxt[N*2],v[N*2];LL c[N*2];
int stack[N],cur[N],father[N];
LL size[N],mark[N];

void addedge(int x,int y,LL z)
{
    ++tot; nxt[tot]=point[x]; point[x]=tot; v[tot]=y; c[tot]=z;
    ++tot; nxt[tot]=point[y]; point[y]=tot; v[tot]=x; c[tot]=z;
}
LL Abs(LL x)
{
    return (x>0)?x:-x;
}
void dfs()
{
    int top=0;
    stack[++top]=1;size[1]=1LL;
    for (int i=1;i<=n;++i) cur[i]=point[i];
    while (top)
    {
        int x=stack[top];
        while (cur[x]&&v[cur[x]]==father[x]) cur[x]=nxt[cur[x]];
        if (!cur[x])
        {
            --top;
            if (father[x])
            {
                size[father[x]]+=size[x];
                ans+=Abs((LL)n-size[x]-size[x])*mark[x];
            }
            continue;
        }
        int vt=v[cur[x]];
        father[vt]=x;size[vt]=1LL;mark[vt]=c[cur[x]];
        stack[++top]=vt;
        cur[x]=nxt[cur[x]];
    }
}

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