splay普通平衡树coedvs4543

题目描述

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

插入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

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

注意细节

#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>using namespace std;const int MAXN=1000001;int size[MAXN],f[MAXN],key[MAXN],cnt[MAXN],ch[MAXN][2];int root = 0,n,sz = 0;void up(int x){    if(x)    {        size[x]=cnt[x];        if(ch[x][1]) size[x]+=size[ch[x][1]];        if(ch[x][0]) size[x]+=size[ch[x][0]];        return;    }}void clear(int x){    key[x]=cnt[x]=f[x]=size[x]=cnt[x]=ch[x][1]=ch[x][0]=0;    return;}bool get(int x){    return ch[f[x]][1]==x;}void rotate(int x){    int y=f[x],z=f[y],k=get(x);    ch[y][k]=ch[x][k^1];    f[ch[y][k]]=y;    ch[x][k^1]=y;    f[x]=z;    f[y]=x;    if(z) ch[z][ch[z][1]==y]=x;    up(y),up(x);    return;}void splay(int x){    for(int fa;fa=f[x];rotate(x))    if(f[fa]) rotate(get(x)==get(fa)?fa:x);    root=x;//别忘了    return;}void insert(int x){    if(root==0)     {        sz++;        size[sz]=cnt[sz]=1;        ch[sz][0]=ch[sz][1]=0;        f[sz]=0;        root=sz;//别忘了        key[sz]=x;        return;    }    int now=root,fa=0;    while(true)    {        if(x==key[now])        {            cnt[now]++;            up(now),up(fa);            splay(now);            break;        }        fa=now;        now=ch[now][key[now]<x];        if(!now)        {            sz++;            ch[sz][0]=ch[sz][1]=0;            f[sz]=fa;            size[sz]=cnt[sz]=1;            ch[fa][key[fa]<x]=sz;//！！！！！别忘了！！！！！！            key[sz]=x;            up(fa);            splay(sz);            break;        }    }    return;}int find(int x){    int now=root,ans=0;    while(true)    {        if(x<key[now]) now=ch[now][0];        else         {            ans+=(ch[now][0]?size[ch[now][0]]:0);            if(x==key[now])            {                splay(now);                return ans+1;            }            ans+=cnt[now];            now=ch[now][1];        }     }}int findx(int x){    int now=root;    while(true)    {        if(ch[now][0] && x<=size[ch[now][0]]) now=ch[now][0];        else         {            int temp=(ch[now][0]?size[ch[now][0]]:0)+cnt[now];            if(x<=temp) return key[now];            x-=temp;            now=ch[now][1];        }    }}int pre(){    int now=ch[root][0];    while(ch[now][1]) now=ch[now][1];    return now;}int nxt(){    int now=ch[root][1];    while(ch[now][0]) now=ch[now][0];    return now;}void del(int x){    int ss=find(x);    if(cnt[root]>1)    {        cnt[root]--;        up(root);        return;    }    if(!ch[root][0] && !ch[root][1])    {        clear(root);        root=0;        return;    }    if(!ch[root][1])     {        int t=root;        root=ch[root][0];        f[root]=0;        clear(t);        return;    }    if(!ch[root][0])    {        int t=root;        root=ch[root][1];        f[root]=0;        clear(t);        return;    }    int ll=pre(),y=root;//!!!!!!    splay(ll);    ch[root][1]=ch[y][1];    f[ch[y][1]]=root;    clear(y);    up(root);    return;}int main(){    int opt,x;    cin>>n;    for(int i=1;i<=n;i++)    {        scanf("%d%d",&opt,&x);        switch(opt)        {            case 1:insert(x);break;            case 2:del(x);break;            case 3:printf("%d\n",find(x));break;            case 4:printf("%d\n",findx(x));break;            case 5:insert(x);printf("%d\n",key[pre()]),del(x);break;            case 6:insert(x);printf("%d\n",key[nxt()]),del(x);break;        }    }    return 0;}