【NEW】[平衡树]模板而已

题目描述

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

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

输入格式：
第一行为n，表示操作的个数，下面n行每行有两个数opt和x，opt表示操作的序号( 1≤opt≤6)

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

输入输出样例

输入样例

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

输出样例

106465
84185
492737
说明

时空限制：1000ms,128M

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

分析

就是平衡树
3注意：
1、能写else就用else，因为&地址操作符会使变量在两条先后的if语句中可能都能进入（即使条件完全相反）
2、左旋旋的是右儿子节点（右旋同理）
3、标总是对的，别去质疑
滑稽

#include <iostream>
#include <cstdio>
#include <ctime>
#include <cstdlib>
#define rep(i,a,b) for (i=a;i<=b;i++)
using namespace std;
struct node
{
    int l,r;
    int key,data,size;
}t[100001];
int tcnt=0,rt=0;

void UPDATE(int x)
{
    t[x].size=t[t[x].r].size+t[t[x].l].size+1;
}

void RIGHT_ROTATE(int &x)
{
    int y=t[x].l;
    t[x].l=t[y].r;t[y].r=x;
    UPDATE(x);UPDATE(y);
    x=y;
}

void LEFT_ROTATE(int &x)
{
    int y=t[x].r;
    t[x].r=t[y].l;t[y].l=x;
    UPDATE(x);UPDATE(y);
    x=y;
}

void INSERT(int &x,int insdata)
{
    if (x==0)
    {
        x=++tcnt;
        t[x].size=1;
        t[x].data=insdata;
        t[x].key=rand();
        return;
    }
    if (insdata<=t[x].data)
    {
        INSERT(t[x].l,insdata);
        if (t[t[x].l].key>t[x].key) RIGHT_ROTATE(x);
    }
    else
    {
        INSERT(t[x].r,insdata);
        if (t[t[x].r].key>t[x].key) LEFT_ROTATE(x);
    }
    UPDATE(x);
}

int GET_RANK(int x,int k)
{
    if (x==0) return 0;
    if (k>t[x].data) return t[t[x].l].size+GET_RANK(t[x].r,k)+1;
    else return GET_RANK(t[x].l,k);
}

int GET_DATA(int x,int k)
{
    if (t[t[x].l].size+1==k) return t[x].data;
    if (k<t[t[x].l].size+1) return GET_DATA(t[x].l,k);
    else return GET_DATA(t[x].r,k-(t[t[x].l].size+1));
}

int GET_PRE(int k)
{
    int ans=-2147483647;
    int x=rt;
    while (x)
    {
        if (k>t[x].data)
        {
            ans=max(ans,t[x].data);
            x=t[x].r;
        }
        else
        x=t[x].l;
    }
    return ans;
}

int GET_NEXT(int k)
{
    int ans=2147483647;
    int x=rt;
    while (x)
    {
        if (k<t[x].data)
        {
            ans=min(ans,t[x].data);
            x=t[x].l;
        }
        else
        x=t[x].r;
    }
    return ans;
}

void DELETE(int &x,int k)
{
    if (x==0) return;
    if (t[x].data==k)
    {
        if (t[x].l||t[x].r)
        {
            if (t[x].l&&(!t[x].r||t[t[x].l].key<t[t[x].r].key))
            {
                RIGHT_ROTATE(x);
                DELETE(t[x].r,k);
            }
            else
            {
                LEFT_ROTATE(x);
                DELETE(t[x].l,k);
            }
            UPDATE(x);
        }
        else x=0;
        return;
    }
    if (k<t[x].data) DELETE(t[x].l,k);
    else DELETE(t[x].r,k);
    UPDATE(x);
}

int main()
{
    int n,i,a,b;
    scanf("%d",&n);
    rep(i,1,n)
    {
        scanf("%d%d",&a,&b);
        if (a==1) INSERT(rt,b);
        if (a==2) DELETE(rt,b);
        if (a==3) printf("%d\n",GET_RANK(rt,b)+1);
        if (a==4) printf("%d\n",GET_DATA(rt,b));
        if (a==5) printf("%d\n",GET_PRE(b));
        if (a==6) printf("%d\n",GET_NEXT(b));
    }
}