[数据结构] 非旋Treap

模板链接：https://www.luogu.org/blog/yhzq/solution-p3369

【模板】普通平衡树（Treap/SBT）

题目描述

您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作：

插入x数

删除x数(若有多个相同的数，因只删除一个)

查询x数的排名(排名定义为比当前数小的数的个数+1。若有多个相同的数，因输出最小的排名)

查询排名为x的数

求x的前驱(前驱定义为小于x，且最大的数)

求x的后继(后继定义为大于x，且最小的数)

输入输出格式

输入格式：

第一行为n，表示操作的个数,下面n行每行有两个数opt和x，opt表示操作的序号( 1 \leq opt \leq 6 1≤opt≤6 )

输出格式：

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

输入输出样例

输入样例#1：

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

输出样例#1：

106465
84185
492737

说明

时空限制：1000ms,128M

1.n的数据范围：n≤100000

2.每个数的数据范围： [-10^7,10^7]

来源：Tyvj1728 原名：普通平衡树

在此鸣谢

题解

膜拜远航之曲大佬

代码

#include<bits/stdc++.h>
using namespace std;
const int M=5e5;
int n,son[M][2],val[M],pri[M],size[M],tot;
void in()
{scanf("%d",&n);}
void up(int v){size[v]=size[son[v][0]]+size[son[v][1]]+1;}
int enrol(int v)
{
    size[++tot]=1;
    val[tot]=v;
    pri[tot]=rand();
    return tot;
}
int merge(int x,int y)
{
    if(!x||!y)return x+y;
    if(pri[x]<pri[y])
    {
        son[x][1]=merge(son[x][1],y);
        up(x);return x;
    }
    else
    {
        son[y][0]=merge(x,son[y][0]);
        up(y);return y;
    }
}
void split(int v,int a,int &x,int &y)
{
    if(!v){x=y=0;return;}
    if(val[v]<=a)x=v,split(son[v][1],a,son[v][1],y);
    else y=v,split(son[v][0],a,x,son[v][0]);
    up(v);
}
int kth(int v,int x)
{
    while(1)
    {
        if(x<=size[son[v][0]])v=son[v][0];
        else 
        if(x==size[son[v][0]]+1)return v;
        else x-=size[son[v][0]]+1,v=son[v][1];
    }
}
void ac()
{
    int a,b,root=0,x,y,z;
    for(int i=1;i<=n;++i)
    {
        scanf("%d%d",&b,&a);
        switch(b)
        {
            case 1:{split(root,a,x,y);root=merge(merge(x,enrol(a)),y);break;}
            case 2:{split(root,a,x,z);split(x,a-1,x,y);y=merge(son[y][0],son[y][1]);root=merge(merge(x,y),z);break;}
            case 3:{split(root,a-1,x,y);printf("%d\n",size[x]+1);root=merge(x,y);break;}
            case 4:{printf("%d\n",val[kth(root,a)]);break;}
            case 5:{split(root,a-1,x,y);printf("%d\n",val[kth(x,size[x])]);root=merge(x,y);break;}
            case 6:{split(root,a,x,y);printf("%d\n",val[kth(y,1)]);root=merge(x,y);break;}
        }
    }
}
int main()
{
    srand(time(0));
    in();ac();
    return 0;
}