UVA12538 Version Controlled IDE (可持久化Treap)

题目链接

维护一个文本，支持：p位置插入串，p位置删除长度c的串，查询历史版本某个位置开始的串。强制在线

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long lld;
const int N = 50100;
char s[1010100];
int on;
struct Node { Node *ch[2]; char c; int sz; }*nill,*root[N];
void up(Node *o) { if (o!=nill) o->sz = o->ch[0]->sz + o->ch[1]->sz +1; }
int ran() { static unsigned ranx = 654123789; return ranx += ranx<<2|1; }
void New_node(Node *&o,char c=0) {
    o = new Node();
    o->ch[0] = o->ch[1] = nill;
    o->c = c; o->sz = 1;
}

// 复制节点，写成函数省事
void copy(Node *&a,Node *b) { 
    New_node(a); 
    a->ch[0] = b->ch[0]; a->ch[1] = b->ch[1]; 
    a->c = b->c; a->sz = b->sz; 
}
bool lor(int a,int b) { return (lld)a*(~1u)>=(lld)ran()*(a+b); } // 注意这里是大于等于，意为(a)/(a+b)的概率返回1

// cut 和 merge 注意不复制nill，保证nill的唯一性
void cut(Node *o,Node *&a,Node *&b,int k) {
    if (k==0) {
        if (o==nill) b = nill;
        else copy(b,o);
        a = nill;
    } else if (k>=o->sz) {
        copy(a,o);
        b = nill;
    } else {
        if (o->ch[0]->sz>=k) {
            copy(b,o);
            cut(o->ch[0],a,b->ch[0],k);
            up(b);
        } else {
            copy(a,o);
            cut(o->ch[1],a->ch[1],b,k - o->ch[0]->sz - 1);
            up(a);
        }
    }
}
void merge(Node *&o,Node *a,Node *b) {
    if (a==nill||b==nill) {
        if (a!=nill) copy(o,a);
        else if (b!=nill) copy(o,b);
        else o = nill;
    } else {
        if (lor(a->sz,b->sz)) {
            copy(o,a);
            merge(o->ch[1],a->ch[1],b);
            up(o);
        } else {
            copy(o,b);
            merge(o->ch[0],a,b->ch[0]);
            up(o);
        }
    }
}
void show(Node *o) {
    if (o==nill) return ;
    show(o->ch[0]);
    printf("%c",o->c);
    if (o->c=='c') on ++;
    show(o->ch[1]);
}
void build(Node *&o,int l,int r) {
    o = nill;
    if (l>r) return ;
    int mid = l+r>>1;
    New_node(o,s[mid]);
    build(o->ch[0],l,mid-1);
    build(o->ch[1],mid+1,r);
    up(o);
}
void insert(Node *&o,Node *pre,int pos) {
    Node *a,*b,*c;
    cut(pre,a,b,pos);
    int len = strlen(s);
    build(c,0,len-1);
    merge(a,a,c);
    merge(o,a,b);
}
void remove(Node *&o,Node *pre,int pos,int len) {
    Node *a,*b,*c;
    cut(pre,a,b,pos-1);
    cut(b,b,c,len);
    merge(o,a,c);
}
void show(Node *o,int pos,int len) {
    Node *a,*b,*c;
    cut(o,a,b,pos-1);
    cut(b,b,c,len);
    show(b);
    puts("");
}
void init() {
    New_node(nill);
    nill->sz = 0;
    for (int i = 0; i < N; i ++) root[i] = nill;
}
int main() {
    int n,vnow,op,p,c,v;
    scanf("%d",&n);
    init();
    while (n--) {
        scanf("%d",&op);
        if (op==1) {
            scanf("%d%s",&p,s);
            insert(root[vnow+1],root[vnow],p-on);
            vnow ++;
        } else if (op==2) {
            scanf("%d%d",&p,&c);
            remove(root[vnow+1],root[vnow],p-on,c-on);
            vnow ++;
        } else {
            scanf("%d%d%d",&v,&p,&c);
            show(root[v-on],p-on,c-on);
        }
    }
    return 0;
}