[BZOJ3223]Tyvj 1729 文藝平衡樹
阿新 • • 發佈:2019-01-31
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;
}