BZOJ 3223 文艺平衡树 (splay)

来源:互联网 发布:电子狗数据更新 编辑:程序博客网 时间:2024/06/04 23:26

题目链接:BZOJ 3223

因为这几天被数据结构虐的各种不忍直视,所以我计划在这一段时间专攻数据结构。但是今天做这道很水的splay还是debug了很久QAQ。感觉这段时间状态好差= =。

#include<cstdio>#include<cstring>#include<iostream>using namespace std;#define maxn (100000+10)int N,M,root,siz=0;int c[maxn][2],tag[maxn],size[maxn],fa[maxn],id[maxn];int read(){int x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}return x*f;}void pushup(int x){int l=c[x][0],r=c[x][1];size[x]=size[l]+size[r]+1;}void pushdown(int x){int l=c[x][0],r=c[x][1];if(tag[x]){swap(c[x][0],c[x][1]);tag[x]=0; tag[r]^=1; tag[l]^=1;}}void rotate(int x,int &k){int y=fa[x],z=fa[y],l,r;if(c[y][0]==x)l=0; else l=1; r=l^1;if(y==k)k=x;else{if(c[z][0]==y)c[z][0]=x;else c[z][1]=x;}fa[c[x][r]]=y; fa[x]=z; fa[y]=x;c[y][l]=c[x][r]; c[x][r]=y;pushup(y); pushup(x);}void splay(int x,int &k){while(x!=k){int y=fa[x],z=fa[y];if(y!=k){if((c[y][0]==x)==(c[z][0]==y))rotate(y,k);else rotate(x,k);}rotate(x,k);}}void built(int l,int r,int f){if(l>r)return ;int now=id[l],last=id[f];if(l==r){fa[now]=last; size[now]=1;if(l<f)c[last][0]=now; else c[last][1]=now;return ;}int mid=(l+r)>>1; now=id[mid];built(l,mid-1,mid); built(mid+1,r,mid);fa[now]=last; pushup(mid);if(mid<f)c[last][0]=now; else c[last][1]=now;}int find(int x,int rk){pushdown(x);int l=c[x][0],r=c[x][1];if(size[l]+1==rk)return x;else if(size[l]>=rk)return find(l,rk);else return find(r,rk-size[l]-1);}void rever(int l,int r){int x,y,z;x=find(root,l); y=find(root,r+2);splay(x,root); splay(y,c[x][1]);z=c[y][0]; tag[z]^=1;}int main(){N=read(); M=read();for(int i=1;i<=N+2;i++)id[i]=++siz;built(1,N+2,0); root=(N+3)>>1;for(int i=1;i<=M;i++){int l=read(),r=read();rever(l,r);}for(int i=2;i<=N+1;i++){printf("%d ",find(root,i)-1);}return 0;}

0 0
原创粉丝点击