【题目描述】
您需要写一种数据结构(可参考题目标题),来维护一些数,其中需要提供以下操作:
1. 插入x数
2. 删除x数(若有多个相同的数,因只删除一个)
3. 查询x数的排名(若有多个相同的数,因输出最小的排名)
4. 查询排名为x的数
5. 求x的前驱(前驱定义为小于x,且最大的数)
6. 求x的后继(后继定义为大于x,且最小的数)
【解题报告】
这道题是一个比较全面的平衡树的题。他的操作比较多样。代码如下:
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
using namespace std;
#define INF 1000000000
struct node{
int v;
int s;
int sz;
node *p;
node *ch[2];
node(){
v=sz=0;
s=-1;
p=ch[0]=ch[1]=this;
}
bool cmp(int s)
{
return this->s<s;
}
int cd(node *o)
{
return this->ch[1]==o?1:0;
}
}Tnull,*null=&Tnull;
void newnode(node *pre,int v,int d)
{
node *u=(node*)malloc(sizeof(node));
u->ch[0]=u->ch[1]=null;
u->sz=1;
u->s=rand();
u->v=v;
u->p=pre;
pre->ch[d]=u;
}
void pushup(node *o)
{
o->sz=o->ch[0]->sz+o->ch[1]->sz+1;
}
void rot(node *o,int d)
{
node *k=o->ch[d];
o->ch[d]=k->ch[d^1];
k->ch[d^1]->p=o;
o->p->ch[o->p->cd(o)]=k;
k->p=o->p;
k->ch[d^1]=o;
o->p=k;
pushup(o);
pushup(k);
}
void Insert(node *o,node *pre,int v,int dd)
{
if(o==null){
newnode(pre,v,dd);
return;
}
int d;
if(v>=o->v){
d=1;
Insert(o->ch[1],o,v,1);
}
else {
d=0;
Insert(o->ch[0],o,v,0);
}
if(o->cmp(o->ch[d]->s))rot(o,d);
else pushup(o);
}
void Delete(node *o,int v)
{
if(o==null)return;
if(o->v==v){
while(o->ch[0]!=null&&o->ch[1]!=null)
{
int d=o->ch[0]->cmp(o->ch[1]->s);
rot(o,d);
--o->p->sz;
}
node *k;
o->ch[0]==null ? k=o->ch[1] : k=o->ch[0];
o->p->ch[o->p->cd(o)]=k;
k->p=o->p;
free(o);
return;
}
v>o->v? Delete(o->ch[1],v) : Delete(o->ch[0],v);
pushup(o);
}
int Find(node *o,int v)
{
if(o==null)return 1;
int ans=0;
if(v<=o->v)ans=Find(o->ch[0],v);
else ans=Find(o->ch[1],v)+o->ch[0]->sz+1;
return ans;
}
int Find2(node *o,int k)
{
if(o==null)return -1;
if(o->ch[0]->sz+1==k)return o->v;
return o->ch[0]->sz>=k? Find2(o->ch[0],k) : Find2(o->ch[1],k-1-o->ch[0]->sz) ;
}
int mini;
int maxi;
void Findp(node *o,int v)
{
if(o==null)return;
if(o->v<v){
if(maxi<o->v)maxi=o->v;
Findp(o->ch[1],v);
}
else Findp(o->ch[0],v);
}
void Findh(node *o,int v)
{
if(o==null)return;
if(o->v>v){
if(mini>o->v)mini=o->v;
Findh(o->ch[0],v);
}
else Findh(o->ch[1],v);
}
void Free(node *o)
{
if(o==null)return;
Free(o->ch[0]);
Free(o->ch[1]);
free(o);
}
int main()
{
int n;
while(scanf("%d",&n)!=EOF)
{
node *root=null;
for(int i=0;i<n;++i)
{
int vis,v;
scanf("%d%d",&vis,&v);
if(vis==1)Insert(root->ch[1],root,v,1);
else if(vis==2)Delete(root->ch[1],v);
else if(vis==3)printf("%d\n",Find(root->ch[1],v));
else if(vis==4)printf("%d\n",Find2(root->ch[1],v));
else if(vis==5)
{
maxi=-INF;
Findp(root->ch[1],v);
printf("%d\n",maxi);
}
else
{
mini=INF;
Findh(root->ch[1],v);
printf("%d\n",mini);
}
}
Free(root->ch[1]);
root->ch[1]=null;
}
return 0;
}