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<cstdio>#include<cstring>#include<iostream>#include<algorithm>#define lson lc[rt],l,mid#define rson rc[rt],mid+1,rusing 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;}