1. 程式人生 > >[BZOJ3223]Tyvj 1729 文藝平衡樹

[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; }