BZOJ 1507 [NOI2003]Editor

title: ‘BZOJ 1507 [NOI2003]Editor’
categories: BZOJ
date: 2016-1-1 01:01:03
tags: [Splay]

---

Description

Input

输入文件editor.in的第一行是指令条数t，以下是需要执行的t个操作。其中： 为了使输入文件便于阅读，Insert操作的字符串中可能会插入一些回车符，请忽略掉它们（如果难以理解这句话，可以参考样例）。除了回车符之外，输入文件的所有字符的ASCII码都在闭区间[32,126]内。且行尾没有空格。 这里我们有如下假定：
MOVE操作不超过50000个，INSERT和DELETE操作的总个数不超过4000，PREV和NEXT操作的总个数不超过200000。
所有INSERT插入的字符数之和不超过2M（1M=1024*1024），正确的输出文件长度不超过3M字节。
DELETE操作和GET操作执行时光标后必然有足够的字符。MOVE、PREV、NEXT操作必然不会试图把光标移动到非法位置。
输入文件没有错误。 对C++选手的提示：经测试，最大的测试数据使用fstream进行输入有可能会比使用stdio慢约1秒。

Output

输出文件editor.out的每行依次对应输入文件中每条GET指令的输出。

Sample

input.txt
15
Insert 26
abcdefghijklmnop
qrstuv wxy
Move 15
Delete 11
Move 5
Insert 1
^
Next
Insert 1
_
Next
Next
Insert 4
.\/.
Get 4
Prev
Insert 1
^
Move 0
Get 22

output.txt
.\/.
abcde^_^f.\/.ghijklmno

Solution

裸的操作，光标放在外面移动就好。
不要用gets，有毒。

Code

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>

#define maxn 2200000+5

using namespace std;

struct Splay_Tree{
    int ch[2];
    int sz; char key;
}tr[maxn],E;

char in[maxn];
int stack[maxn];
int n,root,pos,top,cnt;

void Pushstack(int k){
    tr[k]=E;
    stack[++top]=k;
};

int getpos(){
    return stack[top--];
}

void del(int k){
    if(tr[k].ch[0]) del(tr[k].ch[0]);
    if(tr[k].ch[1]) del(tr[k].ch[1]);
    Pushstack(k);
}

inline int inp(){
    int x=0;
    char ch=getchar();
    while(ch<'0' || ch>'9') ch=getchar();
    while(ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();
    return x;
}

void Update(int k){
    tr[k].sz=tr[tr[k].ch[0]].sz+tr[tr[k].ch[1]].sz+1;
}

void Rotate(int &k,int d){
    int t=tr[k].ch[d^1];
    tr[k].ch[d^1]=tr[t].ch[d]; tr[t].ch[d]=k;
    Update(k); Update(t); k=t;
}

void Splay(int &k,int x){
    int d1=(tr[tr[k].ch[0]].sz<x?1:0),t=tr[k].ch[d1];
    if(d1==1) x-=tr[tr[k].ch[0]].sz+1;
    if(x){
        int d2=(tr[tr[t].ch[0]].sz<x?1:0);
        if(d2==1) x-=tr[tr[t].ch[0]].sz+1;
        if(x){
            Splay(tr[t].ch[d2],x);
            if(d1==d2) Rotate(k,d1^1);
            else Rotate(tr[k].ch[d1],d1);
        }
        Rotate(k,d1^1);
    }
}

void Insert(int l,int rt){
    Splay(root,l+1); Splay(tr[root].ch[1],l+1-tr[tr[root].ch[0]].sz);
    tr[tr[root].ch[1]].ch[0]=rt;
    Update(tr[root].ch[1]); Update(root);
}

void Delete(int l,int r){
    Splay(root,l); Splay(tr[root].ch[1],r+1-tr[tr[root].ch[0]].sz);
    int pos=tr[tr[root].ch[1]].ch[0];
    del(pos);
    tr[tr[root].ch[1]].ch[0]=0;
    Update(tr[root].ch[1]); Update(root);
}

void Output(int k){
    if(tr[k].ch[0]) Output(tr[k].ch[0]);
    printf("%c",tr[k].key);
    if(tr[k].ch[1]) Output(tr[k].ch[1]);
}

void Get(int l,int r){
    Splay(root,l); Splay(tr[root].ch[1],r+1-tr[tr[root].ch[0]].sz);
    int pos=tr[tr[root].ch[1]].ch[0];
    Output(pos);
}

void Build(int l,int r,int &k){
    if(l>r) return;
    if(l==r){
        k=getpos();
        tr[k].ch[0]=tr[k].ch[1]=0;
        tr[k].sz=1; tr[k].key=in[l];
        return;
    }
    int mid=(l+r)>>1; k=getpos();
    Build(l,mid-1,tr[k].ch[0]);
    Build(mid+1,r,tr[k].ch[1]);
    tr[k].key=in[mid];
    Update(k);
}

int main(){
    freopen("1507.in","r",stdin);
    freopen("1507.out","w",stdout);
    n=inp();
    for(int i=maxn-1;i>=1;i--) stack[++top]=i;
    Build(0,1,root);
    for(int i=1;i<=n;i++){
        int x,len=0,rt;
        scanf("%s",in);
        if(in[0]=='I'){
            x=inp();
            while(len<x){
                char c=getchar();
                if(c>=32 && c<=126) in[len++]=c;
            }
            Build(0,x-1,rt); Insert(pos,rt);
        }       
        else if(in[0]=='D'){
            x=inp(); Delete(pos+1,pos+x);
        }
        else if(in[0]=='G'){
            x=inp();
            Get(pos+1,pos+x); puts("");
        }
        else if(in[0]=='M'){
            pos=inp();
            //Splay(root,pos);
        }
        else if(in[0]=='N') pos++;
        else pos--;
    }
    return 0;
}