HiHo 112 Total Highway Distance

转载 2016年08月28日 20:32:46

问题

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

相关文章推荐

Total Highway Distance(贪心)

题目1 : Total Highway Distance 时间限制:10000ms 单点时限:1000ms 内存限制:256MB 描述 Little ...

hihocoder1238 Total Highway Distance(DFS+二分查找/map)

题目链接:点击打开链接 题意描述:给定一棵树,树上每条边有一个权值,记树上任意两点之间路径权值之和为Sx,求所有点对之间Sx之和ans?有下面两种操作 1、修改某条边的权值 2、求ans 解题思路...

hiho一下第65周微软面试题《Highway》题目分析

题意分析 给定一条单行道的高速公路,汽车都是从坐标0,向坐标无穷移动。又因为是单行道,所以后面的车无法超越前面的车。在时刻0时,有 N 辆车同时进入这条单行道,第i辆车从坐标x[i]进入,并且将...

【Leetcode】477. Total Hamming Distance

思路: (1)设置变量oneCount记录某一位上1的个数,result记录总的Hamming Distance。 (2)依次判断32位中的每一位。每判断完一位,右移一位继续判断下一位。 (3)针对...

Leetcode 477. Total Hamming Distance 总海明距离 解题报告

1 解题思想题目的意思就是给了一个数组,现在求总的海明距离,其中: 1、海明距离:任意两个数在二级制的表示下(int = 32bit),每个bit对应的值是1或0,那么这两个数在这32个位置下,取值...
  • MebiuW
  • MebiuW
  • 2016年12月19日 19:04
  • 1576

Leetcode-477. Total Hamming Distance

前言:为了后续的实习面试,开始疯狂刷题,非常欢迎志同道合的朋友一起交流。因为时间比较紧张,目前的规划是先过一遍,写出能想到的最优算法,第二遍再考虑最优或者较优的方法。如有错误欢迎指正。博主首发CSDN...
  • mcf171
  • mcf171
  • 2017年01月01日 18:34
  • 329

LeetCode 477. Total Hamming Distance

477. Total Hamming DistanceThe Hamming distance between two integers is the number of positions at w...

[leetcode, python] Total Hamming Distance 多个数字之间的汉明距离

The Hamming distance between two integers is the number of positions at which the corresponding bits...
  • ysk0825
  • ysk0825
  • 2016年12月30日 01:16
  • 265

vanet-highway

  • 2014年07月12日 17:09
  • 1.44MB
  • 下载

Sicily 1090 highway + Prim + priority_queue

#include #include #include #include using namespace std; class edge{ public: int s,t; edge(...
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:HiHo 112 Total Highway Distance
举报原因:
原因补充:

(最多只允许输入30个字)