您需要写一种数据结构(可参考题目标题),来维护一些数,其中需要提供以下操作:
1.插入数值 x。
2.删除数值 x(若有多个相同的数,应只删除一个)。
3.查询数值 x 的排名(若有多个相同的数,应输出最小的排名)。
4.查询排名为 x 的数值。
5.求数值 x 的前驱(前驱定义为小于 x 的最大的数)。
6.求数值 x 的后继(后继定义为大于 x 的最小的数)。
注意: 数据保证查询的结果一定存在。
输入格式
第一行为 n,表示操作的个数。
接下来 n 行每行有两个数 opt 和 x,opt 表示操作的序号(1≤opt≤6)。
输出格式
对于操作 3,4,5,6 每行输出一个数,表示对应答案。
数据范围
1≤n≤100000,所有数均在 −107 到 107 内。
输入样例:
8
1 10
1 20
1 30
3 20
4 2
2 10
5 25
6 -1
输出样例:
2
20
20
20
思路:平衡树模板
代码:
#include<stdio.h>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N=1e6+10,INF=1e8+10;
int id,root;
struct l
{
int key,val,cnt,size,l,r;
}tr[N];
void push_up(int p)
{
tr[p].size=tr[tr[p].r].size+tr[tr[p].l].size+tr[p].cnt;
}
int get(int key)
{
tr[++id].key=key;
tr[id].val=rand();
tr[id].size=tr[id].cnt=1;
return id;
}
void build()
{
get(-INF),get(INF);
root=1,tr[1].r=2;
push_up(root);
}
void right(int &p)
{
int q=tr[p].l;
tr[p].l=tr[q].r;
tr[q].r=p;
p=q;
push_up(tr[p].r);
push_up(p);
}
void left(int &p)
{
int q=tr[p].r;
tr[p].r=tr[q].l;
tr[q].l=p;
p=q;
push_up(tr[p].l);
push_up(p);
}
void insert(int &p,int key)
{
if(!p) p=get(key);
else if(tr[p].key==key) tr[p].cnt++;
else if(tr[p].key>key)
{
insert(tr[p].l,key);
if(tr[tr[p].l].val>tr[p].val) right(p);
}
else if(tr[p].key<key)
{
insert(tr[p].r,key);
if(tr[tr[p].r].val>tr[p].val) left(p);
}
push_up(p);
}
void remove(int &p,int key)
{
if(!p) return;
else if(tr[p].key==key)
{
if(tr[p].cnt>1) tr[p].cnt--;
else if(tr[p].r||tr[p].l)
{
if(!tr[p].r||tr[tr[p].l].val>tr[tr[p].r].val)
{
right(p);
remove(tr[p].r,key);
}
else
{
left(p);
remove(tr[p].l,key);
}
}
else p=0;
}
else if(tr[p].key>key) remove(tr[p].l,key);
else if(tr[p].key<key) remove(tr[p].r,key);
push_up(p);
}
int get_rank_by_key(int p,int key)
{
if(!p) return 0;
else if(tr[p].key==key) return tr[tr[p].l].size+1;
else if(tr[p].key>key) return get_rank_by_key(tr[p].l,key);
else return tr[tr[p].l].size+tr[p].cnt+get_rank_by_key(tr[p].r,key);
}
int get_key_by_rank(int p,int rank)
{
if(!p) return INF;
else if(tr[tr[p].l].size>=rank) return get_key_by_rank(tr[p].l,rank);
else if(tr[tr[p].l].size+tr[p].cnt>=rank) return tr[p].key;
else return get_key_by_rank(tr[p].r,rank-tr[tr[p].l].size-tr[p].cnt);
}
int get_prex(int p,int key)
{
if(!p) return -INF;
else if(tr[p].key>=key) return get_prex(tr[p].l,key);
return max(tr[p].key,get_prex(tr[p].r,key));
}
int get_next(int p,int key)
{
if(!p) return INF;
else if(tr[p].key<=key) return get_next(tr[p].r,key);
return min(tr[p].key,get_next(tr[p].l,key));
}
int main()
{
int n,m,i,j;
scanf("%d",&n);
ll ans=0;
build();
for(i=0;i<n;i++)
{
int a,x;
scanf("%d",&a);
if(a==1)
{
scanf("%d",&x);
insert(root,x);
}
if(a==2)
{
scanf("%d",&x);
remove(root,x);
}
if(a==3)
{
scanf("%d",&x);
printf("%d\n",get_rank_by_key(root,x)-1);
}
if(a==4)
{
scanf("%d",&x);
printf("%d\n",get_key_by_rank(root,x+1));
}
if(a==5)
{
scanf("%d",&x);
printf("%d\n",get_prex(root,x));
}
if(a==6)
{
scanf("%d",&x);
printf("%d\n",get_next(root,x));
}
}
//printf("%lld\n",ans);
}