hdu4348 To the moon(区间修改,区间查询的主席树)

每个时间建一棵线段树,这样就很好回到过去的某一版本了。然后区间修改就像线段树那样,但是不能下放懒标记,因为你的左右子树可能是公用的历史版本,会出锅。你只能记成绝对标记。每次在路径上累加懒标记。还要注意区间长度。

#include <cstdio>
#include <cstring>
#include <algorithm>
#define ll long long
using namespace std;
int const N=100005;
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-') f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,m,a[N],rt[N],num=0,now=0;
struct node{
    int lc,rc,tag;
    ll sum;
}tree[30*N];
inline void build(int &p,int l,int r){
    p=++num;tree[p].tag=0;
    if(l==r){tree[p].sum=a[l];return;}
    int mid=l+r>>1;
    build(tree[p].lc,l,mid);build(tree[p].rc,mid+1,r);
    tree[p].sum=tree[tree[p].lc].sum+tree[tree[p].rc].sum;
}
inline void change(int &p,int l,int r,int x,int y,int val){
    tree[++num]=tree[p];p=num;tree[p].sum+=(ll)val*(y-x+1);
    if(x==l&&r==y){tree[p].tag+=val;return;}
    int mid=l+r>>1;
    if(y<=mid) change(tree[p].lc,l,mid,x,y,val);
    else if(x>mid) change(tree[p].rc,mid+1,r,x,y,val);
    else change(tree[p].lc,l,mid,x,mid,val),change(tree[p].rc,mid+1,r,mid+1,y,val);
}
ll query(int p,int l,int r,int x,int y){
    if(x==l&&r==y) return tree[p].sum;
    ll res=0;int mid=l+r>>1;
    res+=(ll)tree[p].tag*(y-x+1);//沿途遇到懒标记也是要加上的 
    if(y<=mid) return res+query(tree[p].lc,l,mid,x,y);
    if(x>mid) return res+query(tree[p].rc,mid+1,r,x,y);
    return res+query(tree[p].lc,l,mid,x,mid)+query(tree[p].rc,mid+1,r,mid+1,y);
}
int main(){
//  freopen("a.in","r",stdin);
    while(~scanf("%d%d",&n,&m)){
        num=0;now=0;
        for(int i=1;i<=n;++i) a[i]=read();
        build(rt[0],1,n);
        while(m--){
            char op[5];scanf("%s",op+1);
            if(op[1]=='B'){now=read();continue;}
            int l=read(),r=read();
            if(op[1]=='C'){
                rt[++now]=rt[now-1];
                change(rt[now],1,n,l,r,read());
            }
            if(op[1]=='Q') printf("%lld\n",query(rt[now],1,n,l,r));
            if(op[1]=='H') printf("%lld\n",query(rt[read()],1,n,l,r));
        }
    }return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值