P3369 【模板】普通平衡树

纯板子。。。。

题意：

1. 插入 xx 数
2. 删除 xx 数(若有多个相同的数，因只删除一个)
3. 查询 xx 数的排名(排名定义为比当前数小的数的个数 +1+1 。若有多个相同的数，因输出最小的排名)
4. 查询排名为 xx 的数
5. 求 xx 的前驱(前驱定义为小于 xx ，且最大的数)
6. 求 xx 的后继(后继定义为大于 xx ，且最小的数)

然后。。。

然后就没有然后了。。。。

注意rotate。。。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cctype>
using namespace std;
#define olinr return
#define love_nmr 0
#define INF 0x7ffffffa
#define mod 123123
int n;
int cnt;
int root;
int siz[mod];
int dat[mod];
int fat[mod];
int num[mod];
int son[mod][];
inline int read()
{
    int f=,x=;
    char ch=getchar();
    while(!isdigit(ch))
    {
        if(ch=='-')
            f=-f;
        ch=getchar();
    }
    while(isdigit(ch))
    {
        x=(x<<)+(x<<)+(ch^);
        ch=getchar();
    }
    olinr x*f;
}
inline void put(int x)
{
    if(x<)
    {
        putchar('-');
        x=-x;
    }
    if(x>)
        put(x/);
    putchar(x%+'');
}

/* ------------------------olinr love nmr--------------------------------*/

inline void update(int x)
{
    siz[x]=siz[son[x][]]+siz[son[x][]]+num[x];
}
inline void rotate(int x,int &o)
{
    int y=fat[x];
    int z=fat[y];
    bool xx=son[y][]==x;
    bool yy=xx^;
    if(y==o)
        o=x;
    else
        son[z][son[z][]==y]=x;
    fat[x]=z;
    fat[y]=x;
    fat[son[x][yy]]=y;
    son[y][xx]=son[x][yy];
    son[x][yy]=y;
    update(y);
    update(x);
    olinr;
}
inline void splay(int x,int &o)
{
    while(x!=o)
    {
        int y=fat[x];
        int z=fat[y];
        if(y!=o)
        {
            if((son[y][]==x)^(son[z][]==y))
                rotate(x,o);
            else
                rotate(y,o);
        }
        rotate(x,o);
    }
}
inline int x_rank_n(int x)
{
    int now=root;
    int rank=;
    while(+love_nmr)
    {
        if(x<dat[now])
            now=son[now][];
        else
        {
            rank+=siz[son[now][]];
            if(x==dat[now])
            {
                splay(now,root);
                olinr rank+;
            }
            rank+=num[now];
            now=son[now][];
        }
    }
}
inline int n_rank_x(int x)
{
    int now=root;
    while(+love_nmr)
    {
        if(siz[son[now][]]>=x)
            now=son[now][];
        else
        {
            if(num[now]+siz[son[now][]]>=x)
                olinr dat[now];
            x-=(num[now]+siz[son[now][]]);
            now=son[now][];
        }
    }
}
inline int pre(int x)
{
    x_rank_n(x);
    int now=son[root][];
    while(son[now][])
        now=son[now][];
    olinr now;
}
inline int nxt(int x)
{
    x_rank_n(x);
    int now=son[root][];
    while(son[now][])
        now=son[now][];
    olinr now;
}
inline int del(int x)
{
    int l=pre(x);
    int r=nxt(x);
    splay(l,root);
    splay(r,son[root][]);
    int now=son[son[root][]][];
    siz[now]--;
    num[now]--;
    if(!num[now])
    {
        son[fat[now]][]=;
        fat[now]=;
    }
    splay(son[root][],root);
}
inline void insert(int x)
{
    if(!root)
    {
        root=++cnt;
        siz[root]=;
        dat[root]=x;
        num[root]=;
        olinr;
    }
    int fa=;
    int now=root;
    while(+love_nmr)
    {
        if(dat[now]==x)
        {
            num[now]++;
            siz[now]++;
            splay(now,root);
            olinr;
        }
        fa=now;
        now=son[now][x>dat[now]];
        if(!now)
        {
            now=++cnt;
            fat[now]=fa;
            son[fa][x>dat[fa]]=now;
            num[now]++;
            siz[now]++;
            dat[now]=x;
            splay(now,root);
            olinr;
        }
    }
}
int main()
{
    n=read();
    int flag,x;
    insert(INF);
    insert(-INF);
    for(int i=;i<=n;i++)
    {
        flag=read();
        x=read();
        if(flag==) insert(x);
        if(flag==) del(x);
        if(flag==){put(x_rank_n(x)-);putchar('\n');}
        if(flag==){put(n_rank_x(x+));putchar('\n');}
        if(flag==){insert(x);put(dat[pre(x)]);putchar('\n');del(x);}
        if(flag==){insert(x);put(dat[nxt(x)]);putchar('\n');del(x);}
    }
    olinr love_nmr;
}

注释QAQ

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cctype>
using namespace std;
#define R register
#define IL inline
#define INF 2147483640
#define mod 125125
#define II int
II root;            //根的编号
II n;                 //n个操作
II cnt;                //编号
II fa[mod];            //fa[i]代表i的父亲
II son[mod][];        //son[i][0]代表i的左儿子，son[i][1]代表i的右儿子
II data[mod];        //data[i]代表树上编号为i的点的值
II size[mod];        //size[i]表示以i为根的子树的sum的和 (大小) (包括自己)
II sum[mod];        //sum[i]表示数字i的个数
IL II read()        //快读(有负数)
{
    R II x=,f=;
    char ch=getchar();
    while(!isdigit(ch))
    {
        if(ch=='-')
            f=-f;
        ch=getchar();
    }
    while(isdigit(ch))
    {
        x=(x<<)+(x<<)+(ch^);
        ch=getchar();
    }
    return x*f;
}
IL void update(R II x)  //更新x
{
    size[x]=size[son[x][]]+size[son[x][]]+sum[x];
    //当前点的size == 左儿子的size + 右儿子的size + 当前点的个数
}
IL void rotate(R II x,R II &o)    //旋转
{
    R II y=fa[x];                //y为x的父亲
    R II z=fa[y];                //z为y的父亲，z为x的爷爷
    R bool xx=(son[y][]==x);    //xx表示y的右儿子是否为x
    R bool yy=xx^;                //yy表示y的左儿子是否为x
    if(y==o)                     //x的父亲y是旋转的目标点
        o=x;                    //目标点变为x
    else                        //x的父亲y不是旋转的目标点
        son[z][son    [z][]==y]=x;    //如果z的右儿子是y，那么现在z的右儿子是x；如果z的右儿子不是y，那么现在z的左儿子是x
    fa[x]=z;                     //x的父亲是z
    fa[y]=x;                    //y的父亲是x(x转了上去)
    fa[son[x][yy]]=y;            //如果原来y的左儿子是x，那么现在x的右儿子是y；如果原来y的右儿子是x，那么现在x的左儿子是y
    son[y][xx]=son[x][yy];         //原来x的(右||左)儿子变成了先在y的(左||右)儿子
    son[x][yy]=y;                //如果原来y的(左||右)儿子是x，那么现在x的(右||左)儿子是y
    update(y);                    //维护y
    update(x);                    //维护x
}
IL void splay(R II x,R II &o)
{
    while(x!=o)                                    //只要为转到目标节点
    {
        R II y=fa[x];                            //y是x的父亲
        R II z=fa[y];                            //z是y的父亲，x的爷爷
        if(y!=o)                                //如果y不是目标节点
        {
            if((son[y][]==x)^(son[z][]==y))    //如果xyz不在一条直链上
                rotate(x,o);                    //x朝o的方向转
            else                                //在一条直链上
                rotate(y,o);                    //转y
        }
        rotate(x,o);                            //转x
    }
}
IL void insert(R II x)
{
    if(!root)                            //是根
    {                                    //初始化
        root=++cnt;
        data[root]=x;
        size[root]=;
        sum[root]=;
        return;
    }
    R II now=root;                        //从根开始找合适位置
    R II fat=;                            //now的father
    while()
    {
        if(data[now]==x)                 //树中已有该元素
        {
            sum[now]++;                    //累计数量
            splay(now,root);            //转
            return;
        }
        fat=now;
        now=son[now][data[now]<x];        //now往孩子上跳
        if(!now)                        //跳到了叶子的孩子(还没有)
        {
            now=++cnt;                    //初始化
            fa[now]=fat;
            son[fat][data[fat]<x]=now;
            size[now]++;
            sum[now]++;
            data[now]=x;
            splay(now,root);
            return;
        }
    }
}
IL II x_rank_n(R II x)                 //找树中的数x的排名
{
    R II rank=;
    R II now=root;                    //从根往下找
    while()
    {
        if(x<data[now])                //小于当前点往左跳
            now=son[now][];
        else
        {
            rank+=size[son[now][]];//排名加上左子树大小(x一定比左子树所有值都大，不然就会往左跳)
            if(x==data[now])        //找到了x
            {
                splay(now,root);    //转上去
                return rank+;        //x的位置
            }
            rank+=sum[now];            //往右跳说明它比前面所有数都大，所以要累计前面数的个数
            now=son[now][];        //累计完后在往后跳
        }
    }
}
IL II rank_n_x(R II x)
{
    R II rank=;
    R II now=root;
    while()
    {
        if(size[son[now][]]>=x)                //目标在左子树
            now=son[now][];                    //往左跳
        else
        {
            if(sum[now]+size[son[now][]]>=x)     //now就是x的位置
                return data[now];
            x-=(sum[now]+size[son[now][]]);    //x减左边个数
            now=son[now][];                    //再向右跳
        }
    }
}
IL II pre(R II x)
{
    x_rank_n(x);
    R II now=son[root][];
    while(son[now][])
        now=son[now][];
    return now;
}
IL II nxt(R II x)
{
    x_rank_n(x);
    R II now=son[root][];
    while(son[now][])
        now=son[now][];
    return now;
}
IL void del(R II x)
{
    R II l=pre(x);
    R II r=nxt(x);
    splay(l,root);
    splay(r,son[root][]);
    R II now=son[son[root][]][];
    sum[now]--;
    size[now]--;
    if(!sum[now])
    {
        son[fa[now]][]=;
        fa[now]=;
    }
    splay(son[root][],root);
}
int main()
{
    ios::sync_with_stdio(false);
    n=read();
    insert(-INF);
    insert(INF);
    for(R II i=;i<=n;i++)
    {
        R II x=read();
        R II y=read();
        if(x==) insert(y);
        if(x==) del(y);
        if(x==) printf("%d\n",x_rank_n(y)-);
        if(x==) printf("%d\n",rank_n_x(y+));
        if(x==) insert(y),printf("%d\n",data[pre(y)]),del(y);
        if(x==) insert(y),printf("%d\n",data[nxt(y)]),del(y);
    }
    return ;
}