P3369 【模板】普通平衡树（Treap/SBT）

题目描述

您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作：

1. 插入x数

2. 删除x数(若有多个相同的数，因只删除一个)

3. 查询x数的排名(排名定义为比当前数小的数的个数+1。若有多个相同的数，因输出最小的排名)

4. 查询排名为x的数

5. 求x的前驱(前驱定义为小于x，且最大的数)

6. 求x的后继(后继定义为大于x，且最小的数)

输入输出格式

输入格式：

第一行为n，表示操作的个数,下面n行每行有两个数opt和x，opt表示操作的序号( 1 \leq opt \leq 61≤opt≤6 )

输出格式：

对于操作3,4,5,6每行输出一个数，表示对应答案

输入输出样例

10
1 106465
4 1
1 317721
1 460929
1 644985
1 84185
1 89851
6 81968
1 492737
5 493598

106465
84185
492737

说明

时空限制：1000ms,128M

1.n的数据范围： n \leq 100000n≤100000

2.每个数的数据范围： [-{10}^7, {10}^7][−107,107]

来源：Tyvj1728 原名：普通平衡树

在此鸣谢

code

#include<cstdio>
#include<algorithm>

using namespace std;

const int N = 500100;
int ch
[2],siz
,key
,val
;
int tn,Root;

inline char nc() {
static char buf[100000],*p1 = buf,*p2 = buf;
return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2) ? EOF : *p1++;
}
inline int read() {
int x = 0,f = 1;char ch = getchar();
for (; ch<'0'||ch>'9'; ch = getchar())
if (ch=='-') f = -1;
for (; ch>='0'&&ch<='9'; ch = getchar())
x = x*10+ch-'0';
return x * f;
}
inline void pushup(int x) {
siz[x] = siz[ch[x][0]] + siz[ch[x][1]] + 1;
}
inline int makenode(int x) {
++tn;val[tn] = x;siz[tn] = 1;key[tn] = rand();return tn;
}

int Merge(int x,int y) {
if (!x || !y) return x + y;
if (key[x] < key[y]) {
ch[x][1] = Merge(ch[x][1],y);
pushup(x); return x;
}
else {
ch[y][0] = Merge(x,ch[y][0]);
pushup(y); return y;
}
}
void Split(int now,int k,int &x,int &y) {
if (!now) x = y = 0;
else {
if (val[now] <= k)
x = now,Split(ch[now][1],k,ch[now][1],y);
else
y = now,Split(ch[now][0],k,x,ch[now][0]);
pushup(now);
}
}
inline int getkth(int p,int k) {
while (true) {
if (k == siz[ch
[0]] + 1) return p;
if (ch[p][0] && k <= siz[ch[p][0]]) p = ch[p][0];
else k-= ((ch[p][0] ? siz[ch[p][0]] : 0) + 1),p = ch[p][1];
}
}
int main() {
int x,y,z,opt,k,n = read();
while (n--) {
opt = read(),k = read();
if (opt==1) {
Split(Root,k,x,y);
Root = Merge(Merge(x,makenode(k)),y);
}
else if (opt==2) {
Split(Root,k,x,y);
Split(x,k-1,x,z);
z = Merge(ch[z][0],ch[z][1]);
Root = Merge(Merge(x,z),y);
}
else if (opt==3) {
Split(Root,k-1,x,y);
printf("%d\n",siz[x]+1);
Root = Merge(x,y);
}
else if (opt==4)
printf("%d\n",val[getkth(Root,k)]);
else if (opt==5) {
Split(Root,k-1,x,y);
printf("%d\n",val[getkth(x,siz[x])]);
Root = Merge(x,y);
}
else {
Split(Root,k,x,y);
printf("%d\n",val[getkth(y,1)]);
Root = Merge(x,y);
}
}
return 0;
}