224: Tyvj 1728 普通平衡树
Time Limit: 10 Sec Memory Limit: 128 MBSubmit: 8140 Solved: 3431
[ Submit][ Status][ Discuss]
Description
您需要写一种数据结构(可参考题目标题),来维护一些数,其中需要提供以下操作:
1. 插入x数
2. 删除x数(若有多个相同的数,因只删除一个)
3. 查询x数的排名(若有多个相同的数,因输出最小的排名)
4. 查询排名为x的数
5. 求x的前驱(前驱定义为小于x,且最大的数)
6. 求x的后继(后继定义为大于x,且最小的数)
Input
第一行为n,表示操作的个数,下面n行每行有两个数opt和x,opt表示操作的序号(1<=opt<=6)
Output
对于操作3,4,5,6每行输出一个数,表示对应答案
Sample Input
10
1 106465
4 1
1 317721
1 460929
1 644985
1 84185
1 89851
6 81968
1 492737
5 493598
1 106465
4 1
1 317721
1 460929
1 644985
1 84185
1 89851
6 81968
1 492737
5 493598
Sample Output
106465
84185
492737
84185
492737
HINT
1.n的数据范围:n<=100000
2.每个数的数据范围:[-1e7,1e7]
数据如下http://pan.baidu.com/s/1jHMJwO2
Source
#include<bits/stdc++.h>
using namespace std;
const int maxn = 100010;
int fa[maxn],son[maxn][2];
int siz[maxn],val[maxn],cnt[maxn];
int n;
int rt,tot;
int next,prev;
void updata(int x)
{
siz[x]=siz[son[x][0]]+siz[son[x][1]]+cnt[x];
}
int find(int a,int x)
{
if(a==0) return 0;
if(val[a]==x) return a;
if(x>val[a]) return find(son[a][1],x);
else return find(son[a][0],x);
updata(a);
}
int pre(int x)
{
int y=son[x][0];
while(son[y][1]) y=son[y][1];
return y;
}
void rotate(int x)
{
int y=fa[x],z=fa[y];
int l,r;
if(x==son[y][0]) l=0;else l=1;r=l^1;
if(z)
if(y==son[z][0]) son[z][0]=x;
else son[z][1]=x;
fa[x]=z;fa[y]=x;fa[son[x][r]]=y;
son[y][l]=son[x][r];son[x][r]=y;
updata(y);
}
void splay(int x,int tar)
{
int y,z;
while(fa[x]!=tar)
{
y=fa[x];z=fa[y];
if(z!=tar) if((x==son[y][0])^(y==son[z][0])) rotate(x);else rotate(y);
rotate(x);
}
if(tar==0) rt=x;
updata(x);
}
void insert(int a,int x)
{
if(rt==0)
{
tot++;
rt=tot;
val[tot]=x;
siz[tot]=cnt[tot]=1;
return ;
}
int y;
while(1)
{
if(val[a]==x)
{
cnt[a]++,updata(a);break;
}
int wh;
if(x>val[a]) wh=1;else wh=0;
y=son[a][wh];
if(y==0)
{
tot++;
cnt[tot]=siz[tot]=1;
val[tot]=x;
fa[tot]=a;
son[a][wh]=tot;
a=tot;
break;
}
a=y;
}
splay(a,0);
}
void del(int x)
{
int k=find(rt,x);
if(k==0) return ;
x=k;
splay(x,0);
if(cnt[x]>1){cnt[x]--,updata(x);return;}
if(son[x][0]==0&&son[x][1]==0){rt=0;return;}
if(son[x][0]==0){rt=son[x][1];fa[son[x][1]]=0;return ;}
if(son[x][1]==0){rt=son[x][0];fa[son[x][0]]=0;return ;}
int y=pre(x);
splay(y,x);rt=y;
son[y][1]=son[x][1];fa[y]=0;fa[son[x][1]]=y;updata(y);
}
int rank(int x)
{
int ans=0,now=rt;
while(now)
{
if(val[now]==x) return ans+1+siz[son[now][0]];
else if(val[now]<x) ans+=cnt[now]+siz[son[now][0]],now=son[now][1];
else now=son[now][0];
}
return ans;
}
int kth(int a,int x)
{
if(a==0) return 0;
if(siz[son[a][0]]>=x) return kth(son[a][0],x);
else if(siz[son[a][0]]+cnt[a]>=x) return val[a];
else return kth(son[a][1],x-siz[son[a][0]]-cnt[a]);
}
void lb(int a,int x)
{
if(a==0) return ;
if(val[a]<x) prev=val[a],lb(son[a][1],x);
if(val[a]>=x) lb(son[a][0],x);
}
void ub(int a,int x)
{
if(a==0) return ;
if(val[a]>x) next=val[a],ub(son[a][0],x);
if(val[a]<=x) ub(son[a][1],x);
}
int rlr(int x)
{
if(son[x][0]) rlr(son[x][0]);
printf("%d ",val[x]);
if(son[x][1]) rlr(son[x][1]);
}
int main()
{
scanf("%d",&n);
while(n--)
{
int flag,x;
scanf("%d%d",&flag,&x);
if(flag==1) insert(rt,x);
if(flag==2) del(x);
if(flag==3) printf("%d\n",rank(x));
if(flag==4) printf("%d\n",kth(rt,x));
if(flag==5) lb(rt,x),printf("%d\n",prev);
if(flag==6) ub(rt,x),printf("%d\n",next);
}
return 0;
}
神犇们已经超我一大截了,我不求赶超,只求紧随其后。
int rank(int x)
{
int y=find(rt,x);
splay(y,0);
return siz[son[y][0]];
}