SPOJ-TTM - To the moon(主席树,经典)

题目链接

TTM - To the moon

no tags 

Background

To The Moon is a independent game released in November 2011, it is a role-playing adventure game powered by RPG Maker.

The premise of To The Moon is based around a technology that allows us to permanently reconstruct the memory on dying man. In this problem, we'll give you a chance, to implement the logic behind the scene.

Description

You‘ve been given N integers A[1], A[2],..., A[N]. On these integers, you need to implement the following operations:

• C l r d: Adding a constant d for every {Ai | l <= i <= r}, and increase the timestamp by 1, this is the only operation that will cause the timestamp increase. 
• Q l r: Querying the current sum of {Ai | l <= i <= r}.
• H l r t: Querying a history sum of {Ai | l <= i <= r} in time t.
• B t: Back to time t. And once you decide return to a past, you can never be access to a forward edition anymore.

 .. N, M ≤ 10^5, |A[i]| ≤ 10^9, 1 ≤ l ≤ r ≤ N, |d| ≤ 10^4 .. the system start from time 0, and the first modification is in time 1, t ≥ 0, and won't introduce you to a future state. 

Input

n m
A1 A2 ... An
... (here following the m operations. )

Output

... (for each query, simply print the result. )

Example

Input 1:
10 5
1 2 3 4 5 6 7 8 9 10
Q 4 4
Q 1 10
Q 2 4
C 3 6 3
Q 2 4

Output 1:
4
55
9
15

Input 2:
2 4
0 0
C 1 1 1
C 2 2 -1
Q 1 2
H 1 2 1

Output 2:
0
1

 Submit solution!

题意：

一个长度为n的数组，4种操作 ：

（1）C l r d：区间[l,r]中的数都加1，同时当前的时间戳加1 。

（2）Q l r：查询当前时间戳区间[l,r]中所有数的和 。

（3）H l r t：查询时间戳t区间[l,r]的和 。

（4）B t：将当前时间戳置为t 。时间倒转后就不能再回去。

所有操作均合法 。

题解：

以前写的那几个主席树的题都是属于主席树的应用-查询第k大之类的。

这个题就是主席树的最初用途了。

发现以前写的用于查询第k大的主席树模版写这种题似乎不太合适==

于是换了一种大众一点的写法，可以当作板子。

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<queue>
#include<stack>
using namespace std;
#define rep(i,a,n) for (int i=a;i<n;i++)
#define per(i,a,n) for (int i=n-1;i>=a;i--)
#define pb push_back
#define fi first
#define se second
typedef vector<int> VI;
typedef long long ll;
typedef pair<int,int> PII;
const int maxn=1e6+100;
struct node
{
    int l,r,mark;
    ll sum,tag;
}T[maxn*40];
int root[maxn];
int cnt;//
void pushup(int x)
{
    T[x].sum=T[T[x].l].sum+T[T[x].r].sum;
}
int build(int l,int r)
{
    int root=++cnt;
    T[root].sum=T[root].tag=T[root].mark=0;
    if(l==r)
    {
        scanf("%lld",&T[root].sum);
        T[root].l=T[root].r=0;
        return root;
    }
    int m=(l+r)/2;
    T[root].l=build(l,m);
    T[root].r=build(m+1,r);
    pushup(root);
    return root;
}
void copy(int x,int y)
{
    T[x].sum=T[y].sum;
    T[x].tag=T[y].tag;
    T[x].l=T[y].l;
    T[x].r=T[y].r;
}
void pushdown(int x,int l,int r)
{
    if(T[x].mark)//
    {
        int now=++cnt;
        copy(now,T[x].l);
        T[x].l=now;
        now=++cnt;
        copy(now,T[x].r);
        T[x].r=now;
        T[T[x].l].mark=T[T[x].r].mark=1; //
        T[x].mark=0;
    }
    if(T[x].tag)
    {
        int m=(l+r)/2;
        T[T[x].l].tag+=T[x].tag;
        T[T[x].r].tag+=T[x].tag;
        T[T[x].l].sum+=1ll*T[x].tag*(m-l+1);
        T[T[x].r].sum+=1ll*T[x].tag*(r-m);
        T[x].tag=0;
    }
}
int update(int root,int L,int R,int l,int r,int v)
{
    int now=++cnt;
    if(l==L&&r==R)
    {
        T[now].l=T[root].l;
        T[now].r=T[root].r;  //
        T[now].sum=T[root].sum+1ll*(r-l+1)*v;
        T[now].tag=T[root].tag+v;
        T[now].mark=l==r?0:1;
        return now;
    }
    int m=(L+R)/2;
    T[now].mark=T[now].sum=T[now].tag=0;//
    pushdown(root,L,R);
    if(r<=m)
    {
        T[now].l=update(T[root].l,L,m,l,r,v);
        T[now].r=T[root].r;
    }
    else if(l>m)
    {
        T[now].l=T[root].l;
        T[now].r=update(T[root].r,m+1,R,l,r,v);
    }
    else
    {
        T[now].l=update(T[root].l,L,m,l,m,v);
        T[now].r=update(T[root].r,m+1,R,m+1,r,v);
    }
    pushup(now);
    return now;
}

ll query(int root,int L,int R,int l,int r)
{
    if(L==l&&R==r)
        return T[root].sum;
    pushdown(root,L,R); //
    int m=(L+R)/2;
    if(r<=m) return query(T[root].l,L,m,l,r);
    else if(l>m) return query(T[root].r,m+1,R,l,r);
    else return query(T[root].l,L,m,l,m)+query(T[root].r,m+1,R,m+1,r);
}
int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        int now=0;
        cnt=0;
        root[now]=build(1,n);
        char op[5];
        while(m--)
        {
            scanf("%s",op);
            int l,r,v;
            if(op[0]=='Q')
            {
                scanf("%d%d",&l,&r);
                printf("%lld\n",query(root[now],1,n,l,r));
            }
            else if(op[0]=='C')
            {
                scanf("%d%d%d",&l,&r,&v);
                now++;
                root[now]=update(root[now-1],1,n,l,r,v);
            }
            else if(op[0]=='H')
            {
                scanf("%d%d%d",&l,&r,&v);
                printf("%lld\n",query(root[v],1,n,l,r));
            }
            else if(op[0]=='B')
                scanf("%d",&v),now=v;
        }
        puts("");
    }
    return 0;
}