哗啦啦村的扩建（树形DP）

C - 哗啦啦村的扩建
Time Limit: 2000/1000MS (Java/Others) Memory Limit: 512000/256000KB (Java/Others)
Submit Status
Problem Description

呀呀呀，哗啦啦村在日渐发展中，越来越大了。
唐老师看到这欣欣向荣的情景，感到非常开心。
狗哥在旁边，“喏，我们村子扩建这么快，肯定用了不少钱吧？”
唐老师说：“是呀，不过这些钱都不及我零花钱的万万分之一。”

那么这时候问题来了，唐老师的零花钱至少应该有多少钱呢？
狗哥也想知道这道题的答案，于是他拜托了青君同学，了解到了村子扩建的费用。
啊，原来村子的扩建费用，就是修建道路的费用。
整个村子可以看作有n个房子，村子会修建n-1条道路，保证从任意房子可以到达任意其他房子。
那修建这n-1条道路的费用怎么记呢？对于每条道路，假设这条道路左边有x个房子，右边有y个房子，这条道路长度为k，那么费用就是k*|x-y|。

那么唐老师的零花钱至少有多少钱呢？现在你应该知道了吧。

Input

第一行一个整数，表示这个村子有n个房子
接下来n-1行，表示每条道路的信息，三个整数 a,b,c，表示a,b之间有一条道路，这条路的长度为c
1<=n<=50,000
1≤ai, bi≤n
0 ≤ci≤ 10^6
Output

输出一个整数，表示唐老师的零花钱至少有多少钱
Sample Input

6
1 2 1
1 3 1
1 4 2
6 3 1
5 2 1
Sample Output

2000000000

题目大意：已知，n个房子，n-1条路，保证从任意房子可以到达任意其他房子。对于每条道路，假设这条道路左边有x个房子，右边有y个房子，这条道路长度为k，那么费用就是k*|x-y|。请问，修建这n-1条路的费用是多少？
分析：给出一个最小生成树，自然就想到树形DP，从任意一个房子开始搜都可以。

代码：

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

const int maxn = 100000;

class node{
public:
    int v, w;
    node(int v_, int w_) : v(v_), w(w_) {}
};

vector<node> e[maxn];
long long ans;
int n;
int vis[maxn], size[maxn];

void dfs(int u) {
    size[u] = 1;
    for(int i = 0; i < e[u].size(); i++) {
        node edge = e[u][i];
        if(!vis[edge.v]) {
            vis[edge.v] = 1;
            dfs(edge.v);
            size[u] += size[edge.v];
            ans += (long long)abs(n-size[edge.v]-size[edge.v])*edge.w;
        }
    }
    return;
}

int main() {
    scanf("%d", &n);
    memset(vis, 0, sizeof(vis));
    for(int i = 1; i < n; i++) {
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        e[a].push_back(node(b, c));
        e[b].push_back(node(a, c));
    }
    vis[n] = 1;
    ans = 0;
    dfs(n);
    if(ans != 0) printf("%lld00000000\n", ans);
    else printf("0\n");
    return 0;
}