CDQZ Challenge 2

说在前面：“CDQZ”系列题目数据绝对良心（良苦用心233），提交网址如有需要请私信本蒟蒻。
1002:Challenge 2
查看 提交 统计 提问
总时间限制: 10000ms 单个测试点时间限制: 1000ms 内存限制: 262144kB
描述
给一个空数列，有M次操作，每次操作是以下三种之一：

（1）在数列后加一个数

（2）求数列中某位置的值

（3）撤销掉最后进行的若干次操作（1和3）

输入
第一行一个正整数M。
接下来M行，每行开头是一个字符，若该字符为’A’，则表示一个加数操作，接下来一个整数x，表示在数列后加一个整数x；若该字符为’Q’，则表示一个询问操作，接下来一个整数x，表示求x位置的值；若该字符为’U’，则表示一个撤销操作，接下来一个整数x，表示撤销掉最后进行的若干次操作。
输出
对每一个询问操作单独输出一行，表示答案。
样例输入
9
A 1
A 2
A 3
Q 3
U 1
A 4
Q 3
U 2
Q 3
样例输出
3
4
3
提示
1<=M<=10^5，输入保证合法，且所有整数可用带符号32位整型存储。

题解：可持久化线段树维护数列，分析样例可知此题的“操作”只包括撤销（U）以及加数（A），对于每一棵树记录底层元素（不用pushup()）和元素个数即可。

#include
#include
#include
#include
#define lson lc[rt],l,mid
#define rson rc[rt],mid+1,r
using namespace std;
const int MAXN=100002;
int n,m,tim;
int sum[MAXN*22],lc[MAXN*22],rc[MAXN*22],root[MAXN],num[MAXN];
inline int read() {
    int x=0,f=1;char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1;c=getchar();}
    while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();
    return x*f;
}
void build(int &rt,int l,int r) {
    rt=++tim;
    if (l==r) {
        sum[rt]=0;
        return ;
    }
    int mid=(l+r)>>1;
    build(lson),
    build(rson);
}
void Insert(int pre,int &now,int l,int r,int pos,int val) {
    now=++tim;
    if (l==r) {
        sum[now]=val;
        return ;
    }
    lc[now]=lc[pre],rc[now]=rc[pre];
    int mid=(l+r)>>1;
    if (pos<=mid) Insert(lc[pre],lc[now],l,mid,pos,val);
    else Insert(rc[pre],rc[now],mid+1,r,pos,val);
}
int query(int rt,int l,int r,int pos) {
    if (l==r) return sum[rt];
    int mid=(l+r)>>1;
    if (pos<=mid) return query(lson,pos);
    else return query(rson,pos);
}
int main() {
//  freopen("Challenge 2.in","r",stdin);
    n=MAXN,m=read(),tim=num[0]=0;
    build(root[0],1,n);
    int tree=0,cur=0;
    for (register int i=1;i<=m;++i) {
        char ss;
        while (!isalpha(ss=getchar()));
        if (ss=='A') {
            int val=read();
            ++tree;
            Insert(root[tree-1],root[tree],1,n,num[tree]=++cur,val);
        }
        else if (ss=='Q') {
            int x=read();
            printf("%d\n",query(root[tree],1,n,x));
        }
        else {
            int t=read();
            int last=tree-t;
            ++tree;
            root[tree]=root[last],num[tree]=cur=num[last];
        }
//      printf("%d %d\n",tree,cur);
    }
    return 0;
}