树

题目背景
SOURCE：NOIP2015-HN-CJZX

题目描述
树(tree)是指由 n 个点，n-1 条边构成的连通无向图。如果有一棵树，它的每一条边 (u,v) 都有一个权值 lu,v ，我们把这样的树称作带权树(weighted tree)。

我们知道对于树上的任意两个点，他们之间的路径是唯一的。对于两个点 u，v 来说，我们可以算出 u 与 v 之间的路径上所有边权之和，将其称作 u 与 v 之间路径的长度，记作 d(u,v)。

你的任务是计算：

输入格式:
输入第一行为一个正整数 n 。
接下来 n-1 行，每行三个非负整数 x，y，w，表示点 x 与点 y 之间有一条权值为 w 的一条边。
保证输入的图是一棵树。

输出格式:
输出仅包含一行一个数，即答案。因为结果可能很大，请将答案模 100 000 007 输出。

EG 1
in:　
4
1 2 4
1 3 4
1 4 4
out:
72

EG 2
in:
10
1 2 1320321
2 3 4687651
3 4 1321357
4 5 6651332
5 6 5746513
6 7 5674687
7 8 7451216
8 9 7789965
9 10 8765134
out:
28244404

备注
【样例1说明】
d(1,2)+d(1,3)+d(1,4)=4+4+4=12
d(2,1)+d(2,3)+d(2,4)=4+8+8=20
d(3,1)+d(3,2)+d(3,4)=4+8+8=20
d(4,1)+d(4,2)+d(4,3)=4+8+8=20
所以答案为：12+20+20+20=72。

【数据范围】
共 20 个测试点，其数据特征如下表所示。

对所有数据，有 1≤n≤3×105，0≤w<100 000 007。数据有一定梯度。

---

题解:

代码:

#include<bits/stdc++.h>
using namespace std;

const long long mod=100000007;

struct node
{
    int u,to,next;
    long long w;
}a[600005];

long long l,n;
long long first[300005],son[300005],fa[300005];
long long ans;

inline void add(long long x,long long y,long long z)
{
    a[++l].to=y;
    a[l].u=x;
    a[l].next=first[x];
    a[l].w=z;
    first[x]=l;
}

inline void dfs(int root)
{
    for(int i=first[root];i;i=a[i].next)
        if(a[i].to!=fa[root])
        {
            fa[a[i].to]=root;
            dfs(a[i].to);
            son[root]+=son[a[i].to];
        }
}

int main()
{
    cin>>n;

    for(int i=2;i<=n;++i)
    {
        long long x,y,z;

        cin>>x>>y>>z;

        add(x,y,z);
        add(y,x,z);

        son[i]=1;
    }

    son[1]=1;
    dfs(1);

    for(int i=1;i<=l;++i)
    {
        int o=min(son[a[i].to],son[a[i].u]);
        long long j=(long long)o*(long long)(n-o);

        j*=a[i].w%mod;
        ans+=j%mod;
    }

    cout<<ans%mod<<endl;

    return 0;
}