小明的建设方案

 某国有 n座城市，编号从 1 到 n。小明做为该国的道路工程设计师，打算在各个城市之间建设
n−1 条双向通行的道路，使得任意两个城市都是可以互相到达的。假设建设每i条道路的成本为
??，第 i条道路的长度为 ???????，道路两边连接的城市数分别为 ???? (?) 和 ???? (?) ，则所
有道路的修建总成本 ?为：?? = ??????? ×∣ ???? (?) − ???? (?)∣,
? = ∑?=? ? ?? = ∑?=? ?
??????? ×∣ ???? (?) − ???? (?)∣.
 现在小明设计了一个建设方案，想请你帮他计算一下，该方案的总成本 C。
 输入格式：输入第一行是一个整数 ?（1 ≤ ? ≤ 1,000,000），表示某国的 n座城市。接下来输
入 n−1 行，每行输入三个整数 ?、?、?（1 ≤ ?,? ≤ ?，1 ≤ ? ≤ 1,000,000），表示城市 s 和 t
之间有一条长度为 c 的道路。
 输出格式：输出一行，输出一个整数，表示道路建设的总成本 C。
 样例输入
 5
 1 2 2
 2 3 1
 3 4 4
 3 5 5
 样例输出
 34

初步分析
题目的意思很简单就是求总成本C就行了。注意w的值由那几部分组成，一部分是lengthi。这个好储存，另一部分就是numai和numbi。如果我们对每条边都去通过深度或广度去计算这两个值（及左端子图的节点数和右端子图的节点数）的话。时间复杂度就有O(N^2)数据大的话肯定会卡主的。所以我们把这个图生成一棵树。然后用树形动态规划去计算numa和numb的值。

解题思路
首先我们用f[i]记录结点i为根的子树一共有多少个结点。如果第i条边连接的子节点是x父亲结点是y那么∣ ???? (?) − ???? (?)∣=abs(n-f[x]-f[x])=abs(n-2*f[x]);那么fa[]这个数组的值怎么得到呢很简单。f[i]== ∑? ∈?ℎ????[?]f[j]+1。还需要注意的一点就是我们输入数据的时候直接输入的两个端点。这样最后生成的只是一个无向图，并不是树。所以还需要我们处理一下，把它变成树。如果其中一个的father值不为0，则另一个点的father值就是它，否则相反。

直接上代码

#include <iostream>
#include <string.h>
#include <math.h>
using namespace std;
const int MAX_N=1000001;
int f[MAX_N];
int father[MAX_N];
int n,root=0,ans=0;

struct edge{
    int son;
    int father;
    int lenth;
}eid[MAX_N];

void tree_dp(int node){
    for(int i=1;i<=n;++i){
        if(father[i]==node){
            tree_dp(i);
            f[node]+=f[i];
        }
    }
    f[node]++;
}

int read(){
    int s=0,w=1;
    char ch;
    ch=getchar();
    while (!isdigit(ch)){
        if(ch=='-') w*=-1;
        ch=getchar();
    }
    while(isdigit(ch)){
        s=(s<<1)+(s<<3)+ch-48;
        ch=getchar();
    }
    return s*w;
}

int main() {
    memset(f,0, sizeof(f));
    memset(father,0, sizeof(father));
    n=read();
    for(int i=1;i<=n-1;++i){
        int x,y,lenth;
        x=read();
        y=read();
        lenth=read();
        if(father[x]==0){
            father[x]=y;
            eid[i].son=x;
            eid[i].father=y;
            root=y;}
        else {
            father[y] = x;
            eid[i].son=y;
            eid[i].father=x;
            root = x;
        }
        eid[i].lenth=lenth;
    }

    while(father[root]){
        root=father[root];
    }

    tree_dp(root);

    for(int i=1;i<=n-1;++i){
        ans+=eid[i].lenth*abs(n-2*f[eid[i].son]);
    }

    cout<<ans;
    return 0;
}