JZYZOJ1998 [bzoj3223] 文艺平衡树 splay 平衡树

Posted 2020-10-22 鲸头鹳

http://172.20.6.3/Problem_Show.asp?id=1998

平衡树区间翻转的板子，重新写一遍，给自己码一个板子。

  1 #include<iostream>
  2 #include<cstdio>
  3 #include<cstring>
  4 #include<algorithm>
  5 #include<cmath>
  6 #include<cstdlib>
  7 using namespace std;
  8 const int maxn=100100;
  9 int c[maxn][2],fa[maxn],siz[maxn],rev[maxn];
 10 int n,m,rt;
 11 inline int read(){
 12     int x=0,f=1;char ch=getchar();
 13     while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();}
 14     while(ch>=‘0‘&&ch<=‘9‘){x*=10;x+=ch-‘0‘;ch=getchar();}
 15     return x*f;
 16 }
 17 inline void pushup(int x){//向上总结siz
 18     siz[x]=siz[c[x][0]]+siz[c[x][1]]+1;
 19 }
 20 inline void pushdown(int x){//向下传递rev标记
 21     if(rev[x]){
 22         swap(c[x][0],c[x][1]);
 23         rev[c[x][0]]^=1;rev[c[x][1]]^=1;
 24         rev[x]=0;
 25     }
 26 }
 27 void build(int l,int r,int pa){//不完全二分建树。。
 28     if(l>r)return;
 29     if(l==r){
 30         fa[l]=pa;siz[l]=1;
 31         if(l<pa)c[pa][0]=l;
 32         else c[pa][1]=l;
 33         return;
 34     }
 35     int mid=(l+r)/2;
 36     build(l,mid-1,mid);build(mid+1,r,mid);
 37     if(mid<pa)c[pa][0]=mid;
 38     else c[pa][1]=mid;
 39     fa[mid]=pa;
 40     pushup(mid);
 41 }
 42 int find(int x,int rank){//二分查找排名为rank的数
 43     //因为建树的时候已经默认查找的数在数组中的序号就是它本身
 44     //操作不会改变一个节点的序号，只会改变点和点的联系，从而改变排列顺序
 45     pushdown(x);
 46     //find中的pushdown相当于为后面的splay“开路”
 47     int l=c[x][0],r=c[x][1];
 48     if(siz[l]+1==rank)return x;
 49     else if(siz[l]>=rank)return find(l,rank);
 50     else return find(r,rank-siz[l]-1);
 51 }
 52 void rotate(int x,int &k){//向上旋转
 53     int pa=fa[x];int gpa=fa[pa],l,r;
 54     if(c[pa][0]==x) l=0;
 55     else l=1; r=l^1;//上旋前x在pa的l方向
 56     if(pa==k) k=x;
 57     else{
 58         //如果x旋转不到k，需要关注上旋后和gpa的儿子关系
 59         //观察后面的调用，旋转到k时gpa的儿子关系相当于直接被继承，
 60         if(c[gpa][0]==pa)c[gpa][0]=x;
 61         else c[gpa][1]=x;
 62     }
 63     fa[x]=gpa;fa[pa]=x;fa[c[x][r]]=pa;
 64     c[pa][l]=c[x][r];c[x][r]=pa;
 65     pushup(x);pushup(pa);
 66     
 67 }
 68 void splay(int x,int &k){
 69     while(x!=k){
 70         int pa=fa[x];int gpa=fa[pa];
 71         if(pa!=k){//这里的异或处理了顺位和逆位两种情况
 72             /*顺位先将pa上旋至k，逆位直接将x向上旋，
 73             避免树失衡。*/
 74             if((c[pa][0]==x)^(c[gpa][0]==pa))rotate(x,k);
 75             else rotate(pa,k);
 76         }rotate(x,k);
 77     }
 78 }
 79 void rever(int l,int r){
 80     int x=find(rt,l),y=find(rt,r+2);//找到两个数的位置
 81     splay(x,rt);splay(y,c[x][1]);//将两个数调为根和其rc
 82     //此时，两个数之间的数都在根的rc的lc上
 83     /*    我们开始开始的建树从1到n+2就是为了将1和n+2作为边界
 84     事实上进行操作的是2到n+1这n个数。我们最后取的时候要减1。
 85     所以，翻转l+1到r+1即可。
 86     */
 87     rev[c[y][0]]^=1;
 88 }
 89 int main(){
 90     n=read();m=read();
 91     int x,y;
 92     build(1,n+2,0);rt=(n+3)/2;
 93     for(int i=1;i<=m;i++){
 94         x=read();y=read();
 95         rever(x,y);
 96     }
 97     for(int i=2;i<=n+1;i++){
 98         printf("%d ",find(rt,i)-1);
 99     }
100     return 0;
101 }

View Code