bzoj3224 Tyvj1728 普通平衡树

3224: Tyvj 1728 普通平衡树

Time Limit: 10 Sec Memory Limit: 128 MB

Submit: 7650 Solved: 3223

[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

Sample Output

106465

84185

492737

HINT

1.n的数据范围：n<=100000

2.每个数的数据范围：[-1e7,1e7]

数据如下http://pan.baidu.com/s/1jHMJwO2

Source

平衡树

平衡树模板题
注意Treap的删除操作的写法！！！

#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#define F(i,j,n) for(int i=j;i<=n;i++)
#define D(i,j,n) for(int i=j;i>=n;i--)
#define ll long long
#define N 100005
#define inf 1000000000
using namespace std;
int n,rt,tot;
int rnd
,l
,r
,v
,w
,sz
;
inline int read()
{
int x=0,f=1;char ch=getchar();
while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();}
while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
void lturn(int &k)
{
int tmp=r[k];r[k]=l[tmp];l[tmp]=k;
sz[tmp]=sz[k];sz[k]=sz[l[k]]+sz[r[k]]+w[k];
k=tmp;
}
void rturn(int &k)
{
int tmp=l[k];l[k]=r[tmp];r[tmp]=k;
sz[tmp]=sz[k];sz[k]=sz[l[k]]+sz[r[k]]+w[k];
k=tmp;
}
void insert(int &k,int x)
{
if (!k){k=++tot;v[k]=x;sz[k]=w[k]=1;l[k]=r[k]=0;rnd[k]=rand();return;}
sz[k]++;
if (v[k]==x) w[k]++;
else if (v[k]>x){insert(l[k],x);if (rnd[l[k]]<rnd[k]) rturn(k);}
else{insert(r[k],x);if (rnd[r[k]]<rnd[k]) lturn(k);}
}
void del(int &k,int x)//一定要记住这里面sz[k]--的位置!!!
{
if (v[k]==x)
{
if (w[k]>1) w[k]--,sz[k]--;
else if (!l[k]||!r[k]) k=l[k]+r[k];
else if (rnd[l[k]]<rnd[r[k]]) rturn(k),del(k,x);
else lturn(k),del(k,x);
return;
}
sz[k]--;
if (x<v[k]) del(l[k],x);
else del(r[k],x);
}
int getrnk(int k,int x)
{
if (v[k]==x) return sz[l[k]]+1;
else if (x<v[k]) return getrnk(l[k],x);
else return getrnk(r[k],x)+sz[l[k]]+w[k];
}
int getkth(int k,int x)
{
if (sz[l[k]]<x&&sz[l[k]]+w[k]>=x) return v[k];
else if (x<=sz[l[k]]) return getkth(l[k],x);
else return getkth(r[k],x-sz[l[k]]-w[k]);
}
int pre(int k,int x)
{
if (!k) return -inf;
if (x<=v[k]) return pre(l[k],x);
else
{
int tmp=pre(r[k],x);
return tmp==-inf?v[k]:tmp;
}
}
int suc(int k,int x)
{
if (!k) return inf;
if (x>=v[k]) return suc(r[k],x);
else
{
int tmp=suc(l[k],x);
return tmp==inf?v[k]:tmp;
}
}
int main()
{
srand(233);
n=read();
F(i,1,n)
{
int opt=read(),x=read();
if (opt==1) insert(rt,x);
else if (opt==2) del(rt,x);
else if (opt==3) printf("%d\n",getrnk(rt,x));
else if (opt==4) printf("%d\n",getkth(rt,x));
else if (opt==5) printf("%d\n",pre(rt,x));
else printf("%d\n",suc(rt,x));
}
return 0;
}