Luogu P3391 文艺平衡树(Splay or FHQ Treap)
这道题要求区间反转。。。好东西。。
对于Splay:把l-1旋到根,把r+1旋到根的右儿子,这样r+1的左儿子就是整个区间了,然后对这个区间打个tg
注意要插-Inf和Inf到树里面,防止越界,坐标要+1
#include<cstdio> #include<iostream> #define R register int using namespace std; const int N=100010,Inf=0x3f3f3f3f; inline int g() { R ret=0,fix=1; register char ch; while(!isdigit(ch=getchar())) fix=ch=='-'?-1:fix; do ret=ret*10+(ch^48); while(isdigit(ch=getchar())); return ret*fix; } int n,m,tot,rt; struct node{ int fa,ch[2],sz,tg,vl; #define fa(x) t[x].fa #define ch(x,i) t[x].ch[i] #define sz(x) t[x].sz #define tg(x) t[x].tg #define vl(x) t[x].vl #define ls ch(x,0) #define rs ch(x,1) }t[N]; inline void upd(int x) {sz(x)=sz(ls)+sz(rs)+1;} inline void spread(int x) {if(tg(x)) tg(ls)^=1,tg(rs)^=1,tg(x)=0,swap(ls,rs);} inline void rot(int x) { R y=fa(x),d=ch(y,1)==x; if(fa(y)) ch(fa(y),ch(fa(y),1)==y)=x; fa(x)=fa(y); fa(ch(y,d)=ch(x,d^1))=y; fa(ch(x,d^1)=y)=x; upd(y); } inline void Splay(int x,int f) { while(fa(x)!=f) { R y=fa(x); if(fa(y)!=f) rot((ch(y,1)==x)==(ch(fa(y),1)==y)?y:x); rot(x); } upd(x); if(!f) rt=x; } inline int build(int f,int l,int r) { if(l>r) return 0; R md=l+r>>1,x=++tot; fa(x)=f,++sz(x),vl(x)=md-1; ls=build(x,l,md-1);rs=build(x,md+1,r); upd(x); return x; } inline int get(int pos) { R x=rt; while(1) {spread(x); R s=sz(ls); if(pos==s+1) return x; if(pos<=s) x=ls; else x=rs,pos-=(s+1); } } inline void reverse(int l,int r) { l=get(l),r=get(r+2); Splay(l,0),Splay(r,l); tg(ch(r,0))^=1; } int s; inline void print(int x) { spread(x); if(!x) return ; print(ls); if(vl(x)>=1&&vl(x)<=n) printf("%d ",vl(x)); print(rs); } signed main() { n=g(),m=g(); rt=build(0,1,n+2); for(R i=1;i<=m;++i) {R l=g(),r=g(); reverse(l,r);} print(rt); putchar('\n'); }
对于FHQ Treap,先把[1,l-1]和[l,n]split出来,再把[l,r]和[r+1,n]split出来,在代表区间的子树的根节点打tg,注意这道题的的split是按rank的
不过split完忘了upd。。。我真实沙雕qwq。。。
#include<cstdio> #include<iostream> #include<cstdlib> #define R register int #define ls(x) ch[x][0] #define rs(x) ch[x][1] using namespace std; const int N=100010; inline int g() { R ret=0,fix=1; register char ch; while(!isdigit(ch=getchar())) fix=ch=='-'?-1:fix; do ret=ret*10+(ch^48); while(isdigit(ch=getchar())); return ret*fix; } int n,m,tot,rt; int ch[N][2],sz[N],vl[N],dat[N]; bool tg[N]; inline void upd(int x) {sz[x]=sz[ls(x)]+sz[rs(x)]+1;} inline int cre(int v) {R x=++tot; vl[x]=v,sz[x]=1,dat[x]=rand(); upd(x); return x;} inline void spread(int x) {if(tg[x]) swap(ls(x),rs(x)),tg[ls(x)]^=1,tg[rs(x)]^=1,tg[x]=0;} inline void split(int o,int rk,int& x,int& y) { if(!o) {x=y=0; return ;} spread(o); if(sz[ls(o)]<rk) {x=o; split(rs(o),rk-sz[ls(o)]-1,rs(o),y); upd(x);} else {y=o; split(ls(o),rk,x,ls(o)); upd(y);} } inline int merge(int x,int y) { if(!x||!y) return x+y; if(dat[x]<dat[y]) {spread(x); rs(x)=merge(rs(x),y); upd(x); return x;} else {spread(y); ls(y)=merge(x,ls(y)); upd(y); return y;} } inline void print(int x) {if(!x) return ; spread(x); print(ls(x)); printf("%d ",vl[x]); print(rs(x));} signed main() { srand(100023323); n=g(),m=g(); for(R i=1;i<=n;++i) rt=merge(rt,cre(i)); R x=0,y=0,z=0; for(R i=1;i<=m;++i) { R l=g(),r=g(); split(rt,l-1,x,y); split(y,r-l+1,y,z); tg[y]^=1; rt=merge(x,merge(y,z)); } print(rt); while(1); }
2019.05.06