HDU 4348主席树区间修改

如何区间修改:
首先,我们不能进行线段树上的pushdown操作。因为这样做会让公共节点的值被修改。导致其他版本的线段树在查询时出现错误。

  1. 对于每一次区间修改操作,我们让落在L<l&&r<=R区间上的新开的节点的儿子为历史版本的节点,我们可以把这样的节点理解为分界点。因为修改区间将这段区间覆盖之后,它的孩子节点必然需要新开节点。
  2. 对新开的节点,除了孩子为旧节点的新开节点不需要pushup之外,其他的节点都需要pushup。
  3. 既然不能pushdown,那我们就在query函数中多加一个参数,让它带着lazy继续下传,用来模拟pushdown操作。
#include <bits/stdc++.h>
using namespace std;
#define ll long long

const ll maxn = 1e5+100;

ll n,m,a[maxn];
ll tot,rt[maxn];
struct node{ll ls,rs,sum,lazy;}t[maxn<<4];

inline void pushup(ll rt,ll len){
    t[rt].sum = t[t[rt].ls].sum + t[t[rt].rs].sum + 
        t[t[rt].ls].lazy*(len-len/2) + t[t[rt].rs].lazy*(len/2);
}

ll build(ll l,ll r){
    ll now = ++tot;
    t[now].lazy=0;
    if(l==r){
        t[now].sum = a[l];
        return now;
    }
    ll mid = l+r>>1;
    t[now].ls = build(l,mid);
    t[now].rs = build(mid+1,r);
    t[now].sum = t[t[now].ls].sum + t[t[now].rs].sum;
    return now;
}

ll update(ll rt,ll l,ll r,ll L,ll R,ll k){
    ll now = ++tot;
    t[now] = t[rt];
    if(L<=l&&r<=R){
        t[now].lazy+=k;
        return now;
    }
    ll mid = l+r>>1;
    if(L<=mid) t[now].ls = update(t[rt].ls,l,mid,L,R,k);
    if(mid<R) t[now].rs = update(t[rt].rs,mid+1,r,L,R,k);
    pushup(now,r-l+1);
    return now;
}

ll query(ll rt,ll l,ll r,ll L,ll R,ll laz){
    if(L<=l&&r<=R){
    	//因为不能pushdown,所以某个节点的值为sum+lazy*len
        return t[rt].sum + (t[rt].lazy+laz)*(r-l+1);
    }
    ll mid = l+r>>1,ans=0;
    if(L<=mid)  ans+= query(t[rt].ls,l,mid,L,R,laz+t[rt].lazy);
    if(mid<R) ans+= query(t[rt].rs,mid+1,r,L,R,laz+t[rt].lazy);
    //lazy不能下传,我们让参数带着lazy下传
    return ans;
}

int main(){
    int flag=0;
    while(scanf("%lld%lld",&n,&m)!=EOF){
        if(flag) printf("\n");
        ++flag;
        tot = 0;
        ll now = 0;
        for(ll i=1;i<=n;i++)
            scanf("%lld",&a[i]);
        rt[0] = build(1,n);
        while(m--){
            char op[10];
            ll l,r,t;
            scanf("%s",op);
            if(op[0]=='Q'){
                scanf("%lld%lld",&l,&r);
                printf("%lld\n",query(rt[now],1,n,l,r,0));
            }else if(op[0]=='C'){
                scanf("%lld%lld%lld",&l,&r,&t);
                now++;
                rt[now] = update(rt[now-1],1,n,l,r,t);
            }else if(op[0]=='H'){
                scanf("%lld%lld%lld",&l,&r,&t);
                printf("%lld\n",query(rt[t],1,n,l,r,0));
            }else{
                scanf("%lld",&now);
                tot = rt[now+1];	//节省空间
            }
        }
    }
    return 0;
}
  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值