[BZOJ3223]Tyvj 1729 文艺平衡树

原题地址

Splay练手题.
貌似非旋转式Treap也能做？下次试一下…

AC code:

#include <cstdio>
const int N=100010;
int n,m,cnt;

struct nod{
    bool tr;
    int  v,tot;
    nod  *ch[2],*pr;
}*NIL,pool[N];

struct Splay{
    nod *root;

    Splay(){
        root=NIL=&pool[cnt++];
    }

    void clear(nod *p){
        if(!p->tr) return ;
        nod *t=p->ch[0];p->ch[0]=p->ch[1];p->ch[1]=t;
        p->ch[0]->tr^=1;p->ch[1]->tr^=1;p->tr=0;
    }
    void update(nod *p){
        p->tot=p->ch[0]->tot+p->ch[1]->tot+1;
    }
    void rotate(nod *x,bool t){
        nod *y=x->pr,*z=y->pr,*b=x->ch[t^1];
        b->pr=y;y->pr=x;x->pr=z;y->ch[t]=b;x->ch[t^1]=y;
        if(z->ch[0]==y) z->ch[0]=x;
        else if(z->ch[1]==y) z->ch[1]=x;
        update(y);update(x);
    }
    void splay(nod *x,nod *obj){
        while(x->pr!=obj){
            nod *y=x->pr,*z=y->pr;
            if(z==obj&&y->ch[0]==x) rotate(x,0);
            else if(z==obj&&y->ch[1]==x) rotate(x,1);
            else if(z->ch[0]==y&&y->ch[0]==x){rotate(y,0);rotate(x,0);}
            else if(z->ch[1]==y&&y->ch[1]==x){rotate(y,1);rotate(x,1);}
            else if(z->ch[1]==y){rotate(x,0);rotate(x,1);}
            else{rotate(x,1);rotate(x,0);}
        }
        if(obj==NIL) root=x;
    }
    nod* get(int k){
        nod *p=root;
        clear(p);
        while(p->ch[0]->tot+1!=k){
            if(p->ch[0]->tot+1>k) p=p->ch[0];
            else{
                k-=p->ch[0]->tot+1;
                p=p->ch[1];
            }
            clear(p);
        }
        return p;
    }
    void insert(int k){
        nod **p=&root,*q=NIL;
        while(1){
            if(*p==NIL){
                *p=&pool[cnt++];
                (*p)->v=k;(*p)->tot=1;
                (*p)->pr=q;(*p)->ch[0]=(*p)->ch[1]=NIL;
                splay(*p,NIL);
                break;
            }
            q=*p;
            p=&(*p)->ch[1];
        }
    }
    void reverse(int L,int R){
        if(L==1&&R==n) root->tr^=1;
        else if(L==1){
            splay(get(R+1),NIL);
            root->ch[0]->tr^=1;
        }
        else if(R==n){
            splay(get(L-1),NIL);
            root->ch[1]->tr^=1;
        }
        else{
            splay(get(L-1),NIL);
            splay(get(R+1),root);
            root->ch[1]->ch[0]->tr^=1;
        }
    }
    void LDR(nod *p){
        clear(p);
        if(p->ch[0]!=NIL) LDR(p->ch[0]);
        printf("%d ",p->v);
        if(p->ch[1]!=NIL) LDR(p->ch[1]);
    }
};

int main(){
    Splay T;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) T.insert(i);
    for(int i=1;i<=m;i++){
        int L,R;
        scanf("%d%d",&L,&R);
        T.reverse(L,R);
    }
    T.LDR(T.root);

    return 0;
}