HiHo 112 Total Highway Distance

问题

http://hihocoder.com/contest/hiho112/problem/1

问题分析

题目给定一课最小生成树，求所有节点对之间的距离，然后将所有距离相加之后的和。而且之后要修改一些边的权重，再查询距离。
很容易想到一个方法是：统计每条边会被用到几次，最终答案就是 每条边的权重*用到的次数 然后所有边求和。 由于树的形状是不变的，因此每条边被用到的次数是可以复用。每次修改边的长度后，只需要对被修改的边重新计算，然后更新答案。
而要统计每条边被用几次，可以看做去除当前边后树被分成两部分，两部分节点个数的乘积即为此边被使用的次数。
如果我们使用dfs 遍历整棵树，可以求出每条边所指向的子树对应的节点个数，另外一部分节点的个数即为总节点数- 子树节点数。
另外为了在修改边时方便，我们需要求出每个节点所在的层数， 将子树的节点个数存储在子树的根节点处。当修改时只需要找出要修改的边所对应的子树的根节点。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
enum{maxn= 100006};
vector<int> tree[maxn];
vector<int> treeLen[maxn];
ll total;
int level[maxn];
int nodeNum[maxn];
int path[maxn];
int N, M;
void dfs(int f)
{
    nodeNum[f] = 1;
    for (int i=0; i<tree[f].size(); ++i)
    {
        int v = tree[f][i];
        if (level[v] == 0)
        {
            path[v] = treeLen[f][i]; // 边的长度存储在子树的根节点处。
            level[v] = level[f]+1;
            dfs(v);
            nodeNum[f] += nodeNum[v];
            total += (ll)(N-nodeNum[v]) * (ll)nodeNum[v] * (ll)path[v];
        }
    }
}
void edit(int a, int b, int c)
{
    if (level[a] < level[b])
        swap(a, b);
    int delt = c - path[a];
    path[a] = c;
    total += (ll)delt * (ll)(N-nodeNum[a]) * (ll)nodeNum[a];
}
int main()
{
    scanf("%d %d", &N, &M);
    for (int i=0; i<N-1; ++i)
    {
        int a, b, c;
        scanf("%d %d %d", &a, &b, &c);
        tree[a].push_back(b);
        treeLen[a].push_back(c);
        tree[b].push_back(a);
        treeLen[b].push_back(c);
    }
    total =0;
    memset(level, 0, sizeof(level));
    level[1] = 1;
    dfs(1);
    char q[100];
    for (int i=0; i<M; ++i)
    {
        scanf("%s", q);
        if (q[0] == 'Q')
        {
            printf("%lld\n", total);
        }else
        {
            int a, b, c;
            scanf("%d %d %d", &a, &b, &c);
            edit(a, b, c);
        }
    }
    return 0;
}

注意问题

结果比较大需要用 long long 存储， 答案计算过程中也要转换为long long类型；