hdu 4348 To the moon(主席树区间操作)

题目链接：hdu 4348 To the moon

题意:

给你n个数，有m个操作。

1.给区间[l,r]的所有数+d，并且时间戳+1

2.询问当前时间戳的区间和。

3.询问过去时间戳t的区间和。

4.退回到时间戳t。

题解：

直接上主席树。

不过区间操作的时候push_down空间似乎不是那么够用。

所有把push_down这个操作去掉。

用一个标记记录当前这个区间的累加。

询问的时候就将这个累加传下去。(具体看代码)

最后还有退回状态t的时候可以把cnt=root[t+1],

因为后面的内存已经不会再用了。

#include<bits/stdc++.h>
#define F(i,a,b) for(int i=(a);i<=(b);++i)
using namespace std;
typedef long long ll;

const int N=1e5+7;
struct node{int l,r,lazy;ll sum;}T[N*40];
int n,m,cnt,root[N],now;

void build(int &rt,int l=1,int r=n)
{
    T[rt=++cnt].lazy=0;
    if(l==r){scanf("%lld",&T[rt].sum);return;}
    int m=l+r>>1;
    build(T[rt].l,l,m),build(T[rt].r,m+1,r);
    T[rt].sum=T[T[rt].l].sum+T[T[rt].r].sum;
}

void update(int &x,int y,int L,int R,int v,int l=1,int r=n)
{
    T[x=++cnt]=T[y];
    if(L<=l&&r<=R){T[x].lazy+=v;return;}
    int m=l+r>>1;
    if(L<=m)update(T[x].l,T[y].l,L,R,v,l,m);
    if(R>m)update(T[x].r,T[y].r,L,R,v,m+1,r);
    int ll=max(L,l),rr=min(R,r);
    T[x].sum+=1ll*(rr-ll+1)*v;
}

ll query(int L,int R,int rt,ll add=0,int l=1,int r=n)
{
    if(L<=l&&r<=R)return T[rt].sum+(T[rt].lazy+add)*(r-l+1);
    int m=l+r>>1;ll an=0;
    if(L<=m)an+=query(L,R,T[rt].l,add+T[rt].lazy,l,m);
    if(R>m)an+=query(L,R,T[rt].r,add+T[rt].lazy,m+1,r);
    return an;
}

int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        cnt=0,now=0,build(root[0]);
        F(i,1,m)
        {
            char op[2];
            scanf("%s",op);
            if(op[0]=='C')
            {
                int l,r,d;
                now++;
                scanf("%d%d%d",&l,&r,&d);
                update(root[now],root[now-1],l,r,d);
            }
            else if(op[0]=='Q')
            {
                int l,r;
                scanf("%d%d",&l,&r);
                printf("%lld
",query(l,r,root[now]));
            }
            else if(op[0]=='H')
            {
                int l,r,t;
                scanf("%d%d%d",&l,&r,&t);
                printf("%lld
",query(l,r,root[t]));
            }else scanf("%d",&now),cnt=root[now+1];
        }
    }
    return 0;
}