bzoj 3223 (平衡树)

传送门

题意:

实现某个数列的区间翻转

题解:

获得了一份更为完整的平衡树的板子。

本质上就是在splaysplay的过程中上打一个翻转的lazy taglazy~tag,在输出的或者求第kk大的时候下放翻转标记即可。

代码:

#include <bits/stdc++.h>
using namespace std;
const int maxn=200005;

int ch[maxn][2],par[maxn],val[maxn],cnt[maxn],size[maxn],ncnt,root;
int rev[maxn];
bool chk(int x){
    return ch[par[x]][1]==x;
}
void push_up(int x){
    size[x]=size[ch[x][0]]+size[ch[x][1]]+cnt[x];
}
void rotate(int x){
    int y=par[x],z=par[y],k=chk(x),w=ch[x][k^1];
    ch[y][k]=w; par[w]=y;
    ch[z][chk(y)]=x;par[x]=z;
    ch[x][k^1]=y; par[y]=x;
    push_up(y);
    push_up(x);
}
void push_down(int x){
    if(rev[x]){
        swap(ch[x][1],ch[x][0]);
        rev[ch[x][1]]^=1,rev[ch[x][0]]^=1;
        rev[x]=0;
    }
}
void splay(int x,int goal=0){
    while(par[x]!=goal){
        int y=par[x],z=par[y];
        if(z!=goal){
            if(chk(x)==chk(y)) rotate(y);
            else rotate(x);
        }
        rotate(x);
    }
    if(!goal) root=x;
}

void insert(int x){
    int cur=root,p=0;
    while(cur&&val[cur]!=x){
        p=cur;
        cur=ch[cur][x>val[cur]];
    }
    if(cur){
        cnt[cur]++;
    }
    else{
        cur=++ncnt;
        if(p) ch[p][x>val[p]]=cur;
        ch[cur][0]=ch[cur][1]=0;
        par[cur]=p;val[cur]=x;
        cnt[cur]=size[cur]=1;
    }
    splay(cur);
}


void find(int x){
    int cur=root;
    while(ch[cur][x>val[cur]]&&x!=val[cur]){
        cur=ch[cur][x>val[cur]];
    }
    splay(cur);
}

int kth(int k){
    int cur=root;
    while(1){
        push_down(cur);
        if(ch[cur][0]&&k<=size[ch[cur][0]]){
            cur=ch[cur][0];
        }else if(k>size[ch[cur][0]]+cnt[cur]){
            k-=size[ch[cur][0]]+cnt[cur];
            cur=ch[cur][1];
        }else return cur;
    }
}
int pre(int x){
    find(x);
    if(val[root]<x) return root;
    int cur=ch[root][0];
    while(ch[cur][1]) cur=ch[cur][1];
    return cur;
}

int succ(int x){
    find(x);
    if(val[root]>x) return root;
    int cur=ch[root][1];
    while(ch[cur][0]) cur=ch[cur][0];
    return cur;
}
void reverse(int l,int r){
    int x=kth(l),y=kth(r+2);
    splay(x),splay(y,x);
    rev[ch[y][0]]^=1;
}
void remove(int x){
    int last=pre(x),next=succ(x);
    splay(last);
    splay(next,last);
    int del=ch[next][0];
    if(cnt[del]>1){
        cnt[del]--;
        splay(del);
    }
    else ch[next][0]=0,push_up(next),push_up(root);
}
int n,t;
void output(int x){
    push_down(x);
    if(ch[x][0]) output(ch[x][0]);
    if(val[x]&&val[x]<=n) printf("%d ",val[x]);
    if(ch[x][1]) output(ch[x][1]);
}
int main()
{
    scanf("%d%d",&n,&t);
    for(int i=0;i<=n+1;i++) insert(i);
    while(t--){
        int l,r;
        scanf("%d%d",&l,&r);
        reverse(l,r);
    }
    output(root);
}

posted @ 2019-03-22 09:46  ChenJr  阅读(116)  评论(0编辑  收藏  举报