【HDU - 4348】To the moon（主席树在线区间更新）

BUPT2017 wintertraining(15) #8G

题意

给一个数组a，有n个数，m次操作。\(N, M ≤ 10^5, |A i| ≤ 10^9, 1 ≤ l ≤ r ≤ N, |d| ≤ 10^4\)
操作有四种，
C l r d：更新区间[l,r]，值都加上d，并且时间前进1
Q l r：查询[l,r]的区间和
H l r t：查询t时刻的区间和
B t：回到t时刻（一定是历史时刻），且t时刻之后的操作都作废了。

题解

这道题的关键是怎么更新区间和。
本来线段树更新区间和要push down也就是把懒惰标记下移，但是这样，整个修改区间都需要新建线段树的节点，空间不够。
所以考虑在查询的时候传参数下去，也就是一路加上大区间的lazy值，这样就不用push down了。

代码

#include <cstdio>
#include <cstring>
#define N 100005
#define M (N*25)
typedef long long ll;
namespace PST{
    int T[N],lson[M],rson[M],tot,n;
    ll sum[M], lz[M];
    void init(int _n){
        tot=0;
        n=_n;
        memset(sum,0,sizeof sum);
        memset(lz,0,sizeof lz);
    }
    void push_up(int root,int l,int r){
        sum[root]=sum[lson[root]]+sum[rson[root]]+lz[root]*(r-l+1);
    }
    int build(int l,int r){
        int root=++tot;
        if(l==r){
            scanf("%lld",sum+tot);
            return tot;
        }
        int mid=l+r>>1;
        lson[root]=build(l,mid);
        rson[root]=build(mid+1,r);
        push_up(root,l,r);
        return root;
    }
    int update(int root,int l,int r,int L,int R,int val){
        int newroot=++tot;
        lson[newroot]=lson[root];
        rson[newroot]=rson[root];
        lz[newroot]=lz[root];
        if(L<=l&&r<=R){
            sum[newroot]=sum[root]+(ll)val*(r-l+1);
            lz[newroot]+=val;
            return newroot;
        }
        int mid=l+r>>1;
        if(L<=mid)
            lson[newroot]=update(lson[root],l,mid,L,R,val);
        if(R>mid)
            rson[newroot]=update(rson[root],mid+1,r,L,R,val);
        push_up(newroot,l,r);
        return newroot;
    }
    ll query(int root,int l,int r,int L,int R,ll add){
        if(L<=l&&r<=R){
            return sum[root]+(add)*(r-l+1);
        }
        int mid=l+r>>1;
        ll ans=0;
        if(L<=mid)
            ans+=query(lson[root],l,mid,L,R,lz[root]+add);
        if(R>mid)
            ans+=query(rson[root],mid+1,r,L,R,lz[root]+add);
        return ans;
    }
}
int main(){
    int n,m;
    while(~scanf("%d%d",&n,&m)){
        int now;
        PST::init(n);
        PST::T[now=0]=PST::build(1,n);
        for(int i=0;i<m;++i){
            char op;
            int l,r,t,d;
            scanf(" %c",&op);
            if(op=='Q'){
                scanf("%d%d",&l,&r);
                printf("%lld\n",PST::query(PST::T[now],1,n,l,r,PST::lz[PST::T[now]]));
            }else if(op=='H'){
                scanf("%d%d%d",&l,&r,&t);
                printf("%lld\n",PST::query(PST::T[t],1,n,l,r,PST::lz[PST::T[t]]));
            }else if(op=='B'){
                scanf("%d", &now);
                PST::tot=PST::T[now+1]-1;
            }else if(op=='C'){
                scanf("%d%d%d",&l,&r,&d);
                int tmp=now;
                PST::T[++now]=PST::update(PST::T[tmp],1,n,l,r,d);
            }
        }
    }
}

转载于:https://www.cnblogs.com/flipped/p/HDU4348.html