題目描述

您需要寫一種資料結構（可參考題目標題），來維護一些數，其中需要提供以下操作：

插入 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));
    }
}