P1383 高级打字机

P1383 高级打字机

主席树

一发主席树解决。

插入操作十分显然。

撤销操作复制前面的版本就行。

询问操作十分显然。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define N 1000005
int n,lc[N],rc[N],rt[N],len[N],tot,k;
char a[N],p[3],z[3];
#define mid (l+r)/2
void update(int &o,int pre,int x,int l,int r){
    o=++tot;
    if(l>=r){scanf("%s",z);a[o]=z[0];return ;}
    lc[o]=lc[pre]; rc[o]=rc[pre];
    if(x<=mid) update(lc[o],lc[pre],x,l,mid);
    else update(rc[o],rc[pre],x,mid+1,r);
}
void ask(int o,int x,int l,int r){
    if(l>=r) {printf("%c\n",a[o]);return ;}
    if(x<=mid) ask(lc[o],x,l,mid);
    else ask(rc[o],x,mid+1,r);
}
int main(){
    scanf("%d",&n);
    for(int i=1,q;i<=n;++i){
        scanf("%s",p);
        if(p[0]=='T'){
            ++k;len[k]=len[k-1]+1;
            update(rt[k],rt[k-1],len[k],1,n);
        }else if(p[0]=='Q') scanf("%d",&q),ask(rt[k],q,1,n);
        else if(p[0]=='U'){
            scanf("%d",&q);++k;
            len[k]=len[k-q-1];rt[k]=rt[k-q-1];//复制版本
        }
    }return 0;
}