Wannafly挑战赛4 D.树的距离(dfs序建可持久化线段树)

题意:

在这里插入图片描述

解法:
令dep[x]表示点x到根的距离,
令L[x]和R[x]表示dfs序中x的范围[L,R].

对于查询(x,k),可以变为:[L[x],R[x]]范围内,
所有dep[i]>=dep[x]+k的和,减去cnt*dep[x],cnt是满足条件点的数量
可以将[L[x],R[x]]范围内所有数建立权值线段树做,
因为有多组查询,那么在序列上建立可持久化前缀权值线段树即可.
code:
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define PI pair<ll,int>
const int maxm=1e6+5;
const ll Maa=2e18;
int head[maxm],nt[maxm<<1],to[maxm<<1],w[maxm<<1],cnt;
int lc[maxm*40],rc[maxm*40],num[maxm*40];
ll sum[maxm*40];
int L[maxm],R[maxm],rk[maxm],idx;
int rt[maxm],tot;
ll dep[maxm];
int n,q;
void add(int x,int y,int z){
    cnt++;nt[cnt]=head[x];head[x]=cnt;to[cnt]=y;w[cnt]=z;
}
void dfs(int x,int fa){
    L[x]=++idx;
    rk[idx]=x;
    for(int i=head[x];i;i=nt[i]){
        int v=to[i];
        if(v==fa)continue;
        dep[v]=dep[x]+w[i];
        dfs(v,x);
    }
    R[x]=idx;
}
void update(ll x,ll val,ll l,ll r,int &k,int last){
    k=++tot;
    lc[k]=lc[last];
    rc[k]=rc[last];
    sum[k]=sum[last]+val;
    num[k]=num[last]+1;
    if(l==r)return ;
    ll mid=(l+r)/2;
    if(x<=mid)update(x,val,l,mid,lc[k],lc[last]);
    else update(x,val,mid+1,r,rc[k],rc[last]);
}
PI ask(ll st,ll ed,ll l,ll r,int x,int y){
    if(st<=l&&ed>=r)return {sum[y]-sum[x],num[y]-num[x]};
    ll mid=(l+r)/2;
    PI ans={0LL,0};
    PI temp;
    if(st<=mid){
        temp=ask(st,ed,l,mid,lc[x],lc[y]);
        ans.first+=temp.first;
        ans.second+=temp.second;
    }
    if(ed>mid){
        temp=ask(st,ed,mid+1,r,rc[x],rc[y]);
        ans.first+=temp.first;
        ans.second+=temp.second;
    }
    return ans;
}
signed main(){
    ios::sync_with_stdio(0);
    cin>>n;
    for(int i=1;i<n;i++){
        int p,d;cin>>p>>d;
        add(p,i+1,d);
        add(i+1,p,d);
    }
    dfs(1,1);
    for(int i=1;i<=n;i++){
        update(dep[rk[i]],dep[rk[i]],0,Maa,rt[i],rt[i-1]);
    }
    cin>>q;
    while(q--){
        int x,k;cin>>x>>k;
        int L1=L[x]-1,R1=R[x];
        //查询[L[x],R[x]]中>=dep[x]+k的和
        PI ans=ask(dep[x]+k,Maa,0,Maa,rt[L1],rt[R1]);
        //如果有num个满足条件的点,那么答案需要减去num*dep[x]
        ans.first-=1ll*dep[x]*ans.second;
        cout<<ans.first<<endl;
    }
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值