Splay模板

【bzoj 3224】代码：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<string>
#include<cstdio>
#include<cmath>
#include<cstdlib>
using namespace std;
const int inf=2e9+7;
struct node{
    int x,num,siz;
    node *ch[2],*fa;
    node(int);
    void maintain(){siz=ch[0]->siz+ch[1]->siz+num;}
    int son(){
        if(fa->ch[0]==this) return 0;
        if(fa->ch[1]==this) return 1;
        return -1;
    }
}*null=new node(0);
node:: node(int _){
    x=_;num=siz=null ? 1 : 0;
    fa=ch[0]=ch[1]=null ? null : this;
}
struct Tree{
    node *root,*tmp;
    void ST(){
        root=new node(-inf);
        root->ch[1]=new node(inf);
        root->ch[1]->fa=root;
        root->maintain();
    }
    node* insert(node* &p,int u){
        if(p==null){p=new node(u);return p;}
        node* t;
        if(p->x==u){p->num++;p->siz++;t=p;}
        else if(p->x<u) {t=insert(p->ch[1],u);p->ch[1]->fa=p;}
        else {t=insert(p->ch[0],u);p->ch[0]->fa=p;} 
        p->maintain();return t;
    }
    void Rotate(node* p,bool f){
        node* t=p->ch[f^1];
        p->ch[f^1]=t->ch[f];
        if(t->ch[f]!=null) t->ch[f]->fa=p;
        t->ch[f]=p;
        p->maintain();t->maintain();
        if(~p->son()) p->fa->ch[p->son()]=t;
        t->fa=p->fa;p->fa=t;
    }
    void splay(node* p,bool f){
        if(f){
            while(~p->son()){
                int dir=p->son();
                if(p->fa->son()==dir) Rotate(p->fa->fa,dir^1);
                Rotate(p->fa,dir^1);
            }
            root=p;
        }
        else {
            while(~p->son()){
                int dir=p->son();
                if(p->fa==root) return ;
                if(p->fa->fa==root) {Rotate(p->fa,dir^1);return ;}
                if(p->fa->son()==dir) Rotate(p->fa->fa,dir^1);
                Rotate(p->fa,dir^1);
            }
        }
    }
    node* ch(node* p,int u){
        if(p->x==u) return p;
        if(p->x<u) return ch(p->ch[1],u);
        if(p->x>u) return ch(p->ch[0],u);
    }
    void ch0(node* p,int u){
        if(p==null) return ;
        if(p->x<u) {tmp=p;ch0(p->ch[1],u);}
        else ch0(p->ch[0],u);
    }
    void ch1(node* p,int u){
        if(p==null) return ;
        if(p->x>u) {tmp=p;ch1(p->ch[0],u);}
        else ch1(p->ch[1],u);
    }
    int search(node* p,int sum){
        if(p->ch[0]->siz>=sum) return search(p->ch[0],sum); 
        else if(p->ch[0]->siz+p->num>=sum) return p->x;
        else return search(p->ch[1],sum-p->ch[0]->siz-p->num);
    }
    void solve(int u,int num){
        tmp=null;
        if(num==1){
            tmp=ch(root,u);splay(tmp,1);
            printf("%d\n",root->ch[0]->siz);
        }
        if(num==3){
            ch0(root,u);splay(tmp,1);
            printf("%d\n",root->x);
        }
        if(num==4){
            ch1(root,u);splay(tmp,1);
            printf("%d\n",root->x);
        }
        if(num==2) printf("%d\n",search(root,u+1));
    }
    void Insert(int u){
        node* t=insert(root,u);
        splay(t,1);
    }
    void Del(int u){
        node *r1,*r2;
        tmp=null;ch0(root,u);r1=tmp;
        tmp=null;ch1(root,u);r2=tmp;
        splay(r1,1);splay(r2,0);
        if(root->ch[1]->ch[0]->num>=2) {root->ch[1]->ch[0]->siz--;root->ch[1]->ch[0]->num--;}
        else {delete(root->ch[1]->ch[0]);root->ch[1]->ch[0]=null;}
        root->ch[1]->maintain();root->maintain();
    }
}T;
int n,ai,bi;
int main(){
    T.ST();
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d%d",&ai,&bi);
        if(ai==1) T.Insert(bi);//插入x 
        if(ai==2) T.Del(bi);//删除x 
        if(ai==3) T.solve(bi,1);//查询x数的排名 
        if(ai==4) T.solve(bi,2);//查询排名为x的数 
        if(ai==5) T.solve(bi,3);//求x的前驱 
        if(ai==6) T.solve(bi,4);//求x的后继 
    }
    return 0;
}