[Splay模板大全]Tyvj 1728 普通平衡树

Description

您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作：
1. 插入x数
2. 删除x数(若有多个相同的数，因只删除一个)
3. 查询x数的排名(若有多个相同的数，因输出最小的排名)
4. 查询排名为x的数
5. 求x的前驱(前驱定义为小于x，且最大的数)
6. 求x的后继(后继定义为大于x，且最小的数)

Input

第一行为n，表示操作的个数
下面n行每行有两个数opt和x，opt表示操作的序号(1<=opt<=6)
1.n的数据范围：n<=100000
2.每个数的数据范围：[-2e9,2e9]

Output

对于操作3,4,5,6每行输出一个数，表示对应答案

Sample Input

10
1 106465
4 1
1 317721
1 460929
1 644985
1 84185
1 89851
6 81968
1 492737
5 493598

Sample Output

106465
84185
492737

我本来是不想先用这道题的，但是想想板子的重要性，我还是选择了这道题，此题仅仅是存板子

像这种真的就是套板子的题目，真的没有什么好讲的

这个板子不代表所有（还有恶心的区间操作）

代码：

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define MAXN 1000000
int ch[MAXN][2],f[MAXN],size[MAXN],cnt[MAXN],key[MAXN];
int sz,root;
inline bool get(int x)
{
    return ch[f[x]][1]==x;
}
void update(int x)
{
    if (x){
        size[x]=cnt[x];
        if (ch[x][0]) size[x]+=size[ch[x][0]];
        if (ch[x][1]) size[x]+=size[ch[x][1]];
    }
}
void move(int now)
{
    int fa=f[now],cnt=get(now),faa=f[fa];
    if(fa==root)root=now;
    else if(faa)ch[faa][get(fa)]=now;
    ch[fa][cnt]=ch[now][cnt^1];
    f[ch[now][cnt^1]]=fa;
    ch[now][cnt^1]=fa;
    f[fa]=now;
    f[now]=faa;
    update(fa),update(now);
}
void splay(int x,int &k)
{
    int y,z;
    while(x!=k)
    {
        y=f[x],z=f[y];
        if(y!=k)
        {
            if((ch[y][0]==ch[z][0])^(ch[y][1]==ch[z][1]))move(y);
            else move(x);
        }
        move(x);
    }
}
int find(int x)
{
    int now=root,ans=0;
    while(1)
    {
        if(x<key[now])
            now=ch[now][0];
        else
        {
            ans+=(ch[now][0]?size[ch[now][0]]:0);
            if(x==key[now]){splay(now,root);return ans+1;}
            ans+=cnt[now];
            now=ch[now][1];
        }
    }
}
int findx(int x)
{
     int now=root;
     while(1)
     {
        if(ch[now][0]&&x<=size[ch[now][0]])
            now=ch[now][0];
        else
        {
            int w=(ch[now][0]?size[ch[now][0]]:0)+cnt[now];
            if (x<=w){return key[now];}
            x-=w;now=ch[now][1];
        }
     }
}
void inss(int m)
{
    if(root==0){sz++;root=sz;key[sz]=m;cnt[sz]=1,size[sz]=1;return ;}
    int now=root,fa=0;
    while(1){
        if (m==key[now]){
            cnt[now]++; update(now); update(fa); splay(now,root); break;
        }
        fa=now;
        now=ch[now][key[now]<m];
        if (now==0){
            sz++;
            ch[sz][0]=ch[sz][1]=0;
            f[sz]=fa;
            size[sz]=cnt[sz]=1;
            ch[fa][key[fa]<m]=sz;
            key[sz]=m;
            update(fa);
            splay(sz,root);
            break;
        }
    }
}
int pre()
{
    int now=ch[root][0];
    while (ch[now][1])now=ch[now][1];
    return now;
}
int next()
{
    int now=ch[root][1];
    while (ch[now][0])now=ch[now][0];
    return now;
}
void deel(int x)
{
    int aa=find(x);
    if(cnt[root]>1){cnt[root]--;return ;}
    if (!ch[root][0]&&!ch[root][1]){root=0;return;}
    if(ch[root][0]&&!ch[root][1]){root=ch[root][0];f[root]=0;return ;}
    if(ch[root][1]&&!ch[root][0]){root=ch[root][1];f[root]=0;return ;}
    int old=pre();int rt=root;
    splay(old,root);
    f[ch[rt][1]]=root;
    ch[root][1]=ch[rt][1];
    update(root);
}
int main(){
    int n,opt,x;
    scanf("%d",&n);
    for (int i=1;i<=n;++i){
        scanf("%d%d",&opt,&x);
        switch(opt){
            case 1: inss(x); break;
            case 2: deel(x); break;
            case 3: printf("%d\n",find(x)); break;
            case 4: printf("%d\n",findx(x)); break;
            case 5: inss(x);printf("%d\n",key[pre()]);deel(x);break;
            case 6: inss(x); printf("%d\n",key[next()]);deel(x);break;
        }
    }
}