蒜头君的树

蒜头君的树

题解

先存图，再dfs，dfs的过程中统计以x点为子树的点的总数cnt[x]，这样可以求出来x和fa[x]连的这条边一共被使用了(n-cnt[x])*cnt[x]次
再求出来最短路之和tot（因为是树，所以最短路好求），对于每次修改，只需要将将差乘上使用次数就好了

#include<iostream>
#include<cstdio>
using namespace std;
long long x,n,y,fa[1000001],m,e,head[1000001],cnt[100001],dis[100001];
long long tot;
struct node{long long from,to,cost,nxt;}a[1000001];
void add(long long f,long long t,long long w)
{
    e++;
    a[e].to=t;
    a[e].from=f;
    a[e].cost=w;
    a[e].nxt=head[f];
    head[f]=e;
}
void dfs(long long x)
{
    for(long long i=head[x];i;i=a[i].nxt)
    {
        long long t=a[i].to;
        if(t!=fa[x])
        {
            dfs(t);
            cnt[x]+=cnt[t];
            tot+=(n-cnt[t])*cnt[t]*a[i].cost;
        }
    }
    cnt[x]++;
}
int main()
{
//  freopen("1.txt","r",stdin);

    cin>>n;fa[1]=1;
    for(long long i=1;i<=n-1;i++)
    {
        scanf("%lld%lld",&x,&dis[i+1]);
        fa[i+1]=x;
        add(i+1,x,dis[i+1]);
        add(x,i+1,dis[i+1]);
    }
    dfs(1);
    printf("%lld\n",tot);
    cin>>m;
    for(long long i=1;i<=m;i++)
    {
        scanf("%lld%lld",&x,&y);
        tot+=(y-dis[x])*cnt[x]*(n-cnt[x]);
        dis[x]=y;
        printf("%lld\n",tot);
    }
}

吧啦吧啦吧啦吧啦吧啦吧啦吧啦吧啦吧啦