普通平衡树 bzoj 3224

题目

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

分析

分析个屁
无旋treap——链接

分析

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<vector>
#include<ctime>

#define LL long long
#define mp make_pair  

using namespace std;

const int maxn=100000*2;
typedef pair<int,int> par;

struct tree{
    int lc,rc;
    int data,key;
    int size;
}t[maxn];
int tot;
int root;

void updata(int x)
{
    t[x].size=t[t[x].lc].size+t[t[x].rc].size+1;
}

int merge(int x,int y)
{
    if (x==0||y==0) return x+y;
    if (t[x].key<t[y].key) return t[x].rc=merge(t[x].rc,y),updata(x),x;
    else return t[y].lc=merge(x,t[y].lc),updata(y),y;
}

par split(int x,int n)
{
    if (n==0) return mp(0,x);
    int lc=t[x].lc,rc=t[x].rc;
    if (n==t[lc].size) return t[x].lc=0,updata(x),mp(lc,x);
    if (n==t[lc].size+1) return t[x].rc=0,updata(x),mp(x,rc);
    if (n<t[lc].size)
    {
        par tmp=split(lc,n);
        return t[x].lc=tmp.second,updata(x),mp(tmp.first,x);
    }
    else
    {
        par tmp=split(rc,n-t[lc].size-1);
        return t[x].rc=tmp.first,updata(x),mp(x,tmp.second);
    }
}

int rank(int x,int k)
{
    int ans=0,tmp=(int)1e9; 
    while (k)
    {
        if (x==t[k].data) tmp=min(tmp,ans+t[t[k].lc].size+1);
        if (x>t[k].data) ans+=t[t[k].lc].size+1,k=t[k].rc;
        else k=t[k].lc;
    }
    return tmp==(int)1e9?ans:tmp;
}

int find(int x,int k)
{
    while (true)
    {
        if (t[t[k].lc].size==x-1) return t[k].data;
        if (t[t[k].lc].size>x-1) k=t[k].lc;
        else x=x-t[t[k].lc].size-1,k=t[k].rc;
    }
}

int pre(int x,int k)
{
    int ans=-(int)1e9;
    while (k)
    {
        if (t[k].data<x) ans=max(ans,t[k].data),k=t[k].rc;
        else k=t[k].lc;
    }
    return ans;
}

int net(int x,int k)
{
    int ans=(int)1e9;
    while (k)
    {
        if (t[k].data>x) ans=min(ans,t[k].data),k=t[k].lc;
        else k=t[k].rc;
    }
    return ans;
}

void insert(int x)
{
    int k=rank(x,root);
    par tmp=split(root,k);
    t[++tot].data=x;
    t[tot].key=rand();
    t[tot].size=1;
    root=merge(tmp.first,tot);
    root=merge(root,tmp.second);
}

void del(int x)
{
    int k=rank(x,root);
    par t1=split(root,k);
    par t2=split(t1.first,k-1);
    root=merge(t2.first,t1.second);
}

int n;

int main()
{
    srand((unsigned)time(0));
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
    {
        int op,x;
        scanf("%d%d",&op,&x);
        if(op==1) insert(x);
        else if(op==2) del(x);
        else if(op==3) printf("%d\n",rank(x,root));
        else if(op==4) printf("%d\n",find(x,root));
        else if(op==5) printf("%d\n",pre(x,root));
        else printf("%d\n",net(x,root));
    }
    return 0;
}