计蒜客模拟赛D1T2 蒜头君的树:树上节点之间最短距离和

Posted Leohh

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了计蒜客模拟赛D1T2 蒜头君的树:树上节点之间最短距离和相关的知识,希望对你有一定的参考价值。

题目链接:https://nanti.jisuanke.com/t/16446

题意:

  给你一棵有n个节点的树以及每条边的长度,输出树上节点之间的最短距离和。然后进行m次操作,每次操作更改一条边的长度,分别输出每次操作后树上节点之间的最短距离和。

 

题解:

  最短距离和 = ∑(树上每一条边被最短路经过的次数 * 这条边的长度)

  一个节点到它父节点的边被经过的次数 = 该节点以及它的子孙的节点个数 * 除了该节点和它子孙之外的所有节点总个数

  

  每一个节点以及它子孙节点的个数总和用一遍dfs保存在num数组中,然后算出每个sum[i] = num[i] * (n - num[i]),就可以求出在没有进行任何操作时的最短距离和。

  对于每一次操作将第a个节点到它父节点的边长由原来的len[a]改为b,则将原来的最短距离和dis改为dis + (b - len[a])并输出,同时将len[a]改为b即可。

  预处理&计算操作前的最短距离和的复杂度为O(N),m次询问复杂度O(M),总复杂度为O(N+M)。

  注:本题会爆int,Ctrl+R全换成long long。。。

 

AC Code:

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <vector>
#define MAX_N 100005

using namespace std;

int n,m;
long long dis=0;
long long len[MAX_N];
long long num[MAX_N];
long long sum[MAX_N];
vector<int> edge[MAX_N];

void read()
{
    memset(len,0,sizeof(len));
    cin>>n;
    for(int i=2;i<=n;i++)
    {
        int x,y;
        cin>>x>>y;
        edge[x].push_back(i);
        len[i]=y;
    }
}

long long dfs(int now)
{
    long long tot=1;
    for(int i=0;i<edge[now].size();i++)
    {
        tot+=dfs(edge[now][i]);
    }
    num[now]=tot;
    return tot;
}

void cal_sum()
{
    for(int i=1;i<=n;i++)
    {
        sum[i]=num[i]*(n-num[i]);
    }
}

void cal_dis()
{
    for(int i=1;i<=n;i++)
    {
        dis+=sum[i]*len[i];
    }
}

void solve()
{
    dfs(1);
    cal_sum();
    cal_dis();
    cout<<dis<<endl;
    cin>>m;
    for(int i=0;i<m;i++)
    {
        int a,b;
        cin>>a>>b;
        dis+=(b-len[a])*sum[a];
        len[a]=b;
        cout<<dis<<endl;
    }
}

int main()
{
    read();
    solve();
}

 

以上是关于计蒜客模拟赛D1T2 蒜头君的树:树上节点之间最短距离和的主要内容,如果未能解决你的问题,请参考以下文章

计蒜客 25092 蒜头君的数轴

计蒜客 蒜头君的数轴

61计蒜客 动态规划基础 蒜头君的城堡之旅

计蒜客习题 蒜头君的猜想(埃氏筛)

计蒜客NOIP模拟赛 D1T2 信息传递

计蒜客D2T2 蒜头君的排序(动态维护树状数组)