题目描述
您需要写一种数据结构(可参考题目标题),来维护一些数,其中需要提供以下操作:
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:
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]
【分析】
真 · 模板
http://blog.csdn.net/clove_unique/article/details/50630280
这个博客真的很好懂↑
【代码】
//tyvj 1728
#include<iostream>
#include<cstring>
#include<cstdio>
#define ll long long
#define M(a) memset(a,0,sizeof a)
#define fo(i,j,k) for(i=j;i<=k;i++)
using namespace std;
const int mxn=100005;
int n,m,sz,opt,root;
int f[mxn],ch[mxn][2],key[mxn],cnt[mxn],size[mxn];
inline void clear(int x)
{
f[x]=ch[x][0]=ch[x][1]=key[x]=cnt[x]=size[x]=0;
}
inline int get(int x) //左子树or右子树
{
if(ch[f[x]][0]==x) return 0;
return 1;
}
inline void update(int x) //更新size
{
if(!x) return;
size[x]=cnt[x];
if(ch[x][0]) size[x]+=size[ch[x][0]];
if(ch[x][1]) size[x]+=size[ch[x][1]];
}
inline void rotate(int x)
{
int old=f[x],elder=f[old],which=get(x);
ch[old][which]=ch[x][which^1],f[ch[x][which^1]]=old;
ch[x][which^1]=old,f[old]=x;
f[x]=elder;
if(elder) ch[elder][ch[elder][1]==old]=x;
update(old),update(x);
}
inline void splay(int x)
{
for(int fa;(fa=f[x]);rotate(x))
if(f[fa]) rotate((get(x)==get(fa)?fa:x));
root=x;
}
inline void insert(int x)
{
if(!root)
{
root=(++sz);
ch[sz][0]=ch[sz][1]=f[sz]=0;
key[sz]=x,cnt[sz]=1;
return;
}
int now=root,fa=0;
while(1)
{
if(key[now]==x)
{
cnt[now]++;
update(now),update(fa);
splay(now);
return;
}
fa=now;now=ch[now][x>key[now]];
if(now==0)
{
sz++;
ch[sz][0]=ch[sz][1]=0,size[sz]=cnt[sz]=1,key[sz]=x;
f[sz]=fa,ch[fa][x>key[fa]]=sz;
update(fa);
splay(sz);
return;
}
}
}
inline int find(int x) //x的排名
{
int ans=1,now=root;
while(1)
{
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;}
ans+=cnt[now];
now=ch[now][1];
}
}
}
inline int number(int x)
{
int now=root;
while(1)
{
if(ch[now][0] && x<=size[ch[now][0]]) now=ch[now][0];
else
{
int tmp=(ch[now][0]?size[ch[now][0]]:0)+cnt[now];
if(x<=tmp) return key[now];
x-=tmp;now=ch[now][1];
}
}
}
inline int pre()
{
int now=ch[root][0];
while(ch[now][1]) now=ch[now][1];
return now;
}
inline int next()
{
int now=ch[root][1];
while(ch[now][0]) now=ch[now][0];
return now;
}
inline void del(int x)
{
find(x);
if(cnt[root]>1) {cnt[root]--;return;}
//1.only one point
if(!ch[root][0] && !ch[root][1]) {clear(root);root=0;return;}
//2.no left child
if(!ch[root][0]) {int old=root;root=ch[root][1];clear(old);f[root]=0;return;}
//3.no right child
if(!ch[root][1]) {int old=root;root=ch[root][0];clear(old);f[root]=0;return;}
//4.two children
int left=pre(),old=root;
splay(left);
f[ch[old][1]]=root;
ch[root][1]=ch[old][1];
clear(old);
update(root);
}
int main()
{
int i,j,x;
scanf("%d",&n);
fo(i,1,n)
{
scanf("%d%d",&opt,&x);
if(opt==1) insert(x);
if(opt==2) del(x);
if(opt==3) printf("%d\n",find(x));
if(opt==4) printf("%d\n",number(x));
if(opt==5) insert(x),printf("%d\n",key[pre()]),del(x);
if(opt==6) insert(x),printf("%d\n",key[next()]),del(x);
}
// printf("haha\n");
return 0;
}
//2
//1 106465
//1 317721