关闭

HiHo 112 Total Highway Distance

90人阅读 评论(0) 收藏 举报
分类:

问题

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类型;

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:21224次
    • 积分:1356
    • 等级:
    • 排名:千里之外
    • 原创:108篇
    • 转载:35篇
    • 译文:4篇
    • 评论:0条
    文章分类