P4008 [NOI2003]文本编辑器

思路

FHQ Treap的板子
用FHQ Treap维护中序遍历序列即可
然后数组开够！

代码

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
struct Node{
    int lson,rson,sz,num,ran;
}FHQ[300000*30];
int nowpos,n,m,Nodecnt,a[300000*30],root;
int new_Node(int num){
    ++Nodecnt;
    FHQ[Nodecnt].lson=FHQ[Nodecnt].rson=0;
    FHQ[Nodecnt].ran=rand();
    FHQ[Nodecnt].num=num;
    FHQ[Nodecnt].sz=1;
    return Nodecnt;
}
void pushup(int o){
    FHQ[o].sz=FHQ[FHQ[o].lson].sz+FHQ[FHQ[o].rson].sz+1;
}
void split(int val,int now,int &x,int &y){//sz
    if(!now){
        x=y=0;
        return;
    }
    if(val<=FHQ[FHQ[now].lson].sz){
        y=now;
        split(val,FHQ[now].lson,x,FHQ[now].lson);
    }
    else{
        x=now;
        split(val-FHQ[FHQ[now].lson].sz-1,FHQ[now].rson,FHQ[now].rson,y);
    }
    pushup(now);
}
int merge(int x,int y){//x.w<y.w
    if(x*y==0)
        return x+y;
    if(FHQ[x].ran<FHQ[y].ran){
        FHQ[x].rson=merge(FHQ[x].rson,y);
        pushup(x);
        return x;
    }
    else{
        FHQ[y].lson=merge(x,FHQ[y].lson);
        pushup(y);
        return y;
    }
}
int build(int l,int r){
    if(l>r)
        return 0;
    int mid=(l+r)>>1;
    int t=new_Node(a[mid]);
    FHQ[t].lson=build(l,mid-1);
    FHQ[t].rson=build(mid+1,r);
    pushup(t);
    return t;
}
void query(int o){
    if(!o)
        return;
    query(FHQ[o].lson);
    printf("%c",FHQ[o].num);
    query(FHQ[o].rson);
}
char opt[20];
int main(){
    root=new_Node(-1);
    nowpos=1;
    scanf("%d",&m);
    for(int i=1;i<=m;i++){
        scanf("%s",opt);
        if(opt[0]=='M'){
            int x;
            scanf("%d",&x);
            nowpos=x+1;
        }
        else if(opt[0]=='I'){
            int lroot,rroot,newroot;
            split(nowpos,root,lroot,rroot);
            scanf("%d",&n);
            int cnt=1;
            char c=getchar();
            while(cnt<=n){
                c=getchar();
                if(c>=32&&c<=126){
                    a[cnt]=c;
                    cnt++;
                }
            }
            newroot=build(1,n);
            root=merge(merge(lroot,newroot),rroot);
        }
        else if(opt[0]=='D'){
            int lroot,tmproot,midroot,rroot;
            scanf("%d",&n);
            split(nowpos,root,lroot,tmproot);
            split(n,tmproot,midroot,rroot);
            root=merge(lroot,rroot);
        }
        else if(opt[0]=='G'){
            int lroot,tmproot,midroot,rroot;
            scanf("%d",&n);
            split(nowpos,root,lroot,tmproot);
            split(n,tmproot,midroot,rroot);
            query(midroot);
            printf("\n");
            root=merge(lroot,merge(midroot,rroot));
        }
        else if(opt[0]=='P'){
            nowpos--;
        }
        else if(opt[0]=='N'){
            nowpos++;
        }
    }
    return 0;
}