题目描述
您需要写一种数据结构(可参考题目标题),来维护一些数,其中需要提供以下操作:
插入x数
删除x数(若有多个相同的数,因只删除一个)
查询x数的排名(若有多个相同的数,因输出最小的排名)
查询排名为x的数
求x的前驱(前驱定义为小于x,且最大的数)
求x的后继(后继定义为大于x,且最小的数)
输入输出格式
输入格式:
第一行为n,表示操作的个数,下面n行每行有两个数opt和x,opt表示操作的序号(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.每个数的数据范围:[-1e7,1e7]
来源:Tyvj1728 原名:普通平衡树
在此鸣谢
//我们用SPlay 来写Treap的模板题 谢徐队的代码参考
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define MAXN 1000000
int cnt[MAXN],key[MAXN],ch[MAXN][2],fa[MAXN],siz[MAXN];
int sz,root;
inline void UpDate(int u){
siz[u]=cnt[u];
if(ch[u][0]) siz[u]+=siz[ch[u][0]];
if(ch[u][1]) siz[u]+=siz[ch[u][1]];
}
inline int pre(){
int u=ch[root][0];
while(ch[u][1]) u=ch[u][1];//宸﹀瀛愮殑鍙冲瀛愮殑鍙?..
return u;
}
inline int next(){
int u=ch[root][1];
while(ch[u][0]) u=ch[u][0];
return u;
}
inline int getson(int x){ return ch[fa[x]][1]==x; }
inline void clear(int x){
ch[x][0]=ch[x][1]=cnt[x]=siz[x]=fa[x]=key[x]=0;
}
inline void Roteta(int x){
int f=fa[x],f_fa=fa[f],k=getson(x);
ch[f][k]=ch[x][k^1];fa[ch[f][k]]=f;
ch[x][k^1]=f;fa[f]=x;
fa[x]=f_fa;
if(f_fa) ch[f_fa][ch[f_fa][1]==f]=x;
UpDate(f);UpDate(x);
}
inline void SPlay(int x){
for(int f;f=fa[x];Roteta(x))
if(fa[f]) Roteta(getson(x)==getson(f) ? f :x);
root=x;
}
inline int findrank(int x){//鏌ヨx鐨勬帓鍚?
int now=root,ans=0;
while(1){
if(x<key[now]) now=ch[now][0];
else{
ans+=ch[now][0] ? siz[ch[now][0]] : 0;
if(x==key[now]){
SPlay(now);
return ans+1;
}
ans+=cnt[now];now=ch[now][1];
}
}
}
inline int findx(int x){
int now=root;
while(1){
if(ch[now][0]&&x<=siz[ch[now][0]]) now=ch[now][0];
else{
int temp=(ch[now][0] ? siz[ch[now][0]] : 0)+cnt[now];
if(x<=temp) return key[now];
x-=temp;
now=ch[now][1];
}
}
}
inline void create(int x){
sz++;key[sz]=x;
cnt[sz]=siz[sz]=1;
ch[sz][0]=ch[sz][1]=fa[sz]=0;
}
void insert(int x){
if(!root) create(x),root=sz;
else{
int now=root,f=0;
while(1){
if(key[now]==x){
cnt[now]++;siz[now]++;
SPlay(now);break;
}
f=now;now=ch[f][x>key[f]];
if(!now){
create(x);
fa[sz]=f;
ch[f][x>key[f]]=sz;
SPlay(sz);
break;
}
}
}
}
inline void del(int x){
int t=findrank(x);//x鐨勬帓鍚?
if(cnt[root]>1){
cnt[root]--;siz[root]--;
return ;
}
if(!ch[root][0]&&!ch[root][1]){ clear(root);root=0;return; }
if(!ch[root][0]){
int cur=root;
root=ch[root][1];
fa[root]=0;
clear(cur);
return;
}
if(!ch[root][1]){
int tmp=root;
root=ch[root][0];
fa[root]=0;
clear(tmp);
return;
}
int pre1=pre(),tmp=root;
SPlay(pre1);
ch[root][1]=ch[tmp][1];
fa[ch[tmp][1]]=root;
clear(tmp);
UpDate(root);
}
int main(){
int n,opt,x;
scanf("%d",&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",findrank(x));break;//x鐨勬帓鍚?
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[next()]);del(x);break;
}
}
return 0;
}