P3369 模板普通平衡树FHQtreap

Posted 2020-10-17 HWIM

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每行输出一个数，表示对应答案

 

输入输出样例

输入样例#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 \leq 100000n≤100000

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

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

在此鸣谢

 

 

code

 1 #include<cstdio>
 2 #include<algorithm>
 3 
 4 using namespace std;
 5 
 6 const int N = 500100;
 7 int ch[N][2],siz[N],key[N],val[N];
 8 int tn,Root;
 9 
10 inline char nc() {
11     static char buf[100000],*p1 = buf,*p2 = buf;
12     return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2) ? EOF : *p1++;
13 }
14 inline int read() {
15     int x = 0,f = 1;char ch = getchar();
16     for (; ch<‘0‘||ch>‘9‘; ch = getchar()) 
17         if (ch==‘-‘) f = -1;
18     for (; ch>=‘0‘&&ch<=‘9‘; ch = getchar()) 
19         x = x*10+ch-‘0‘;
20     return x * f;
21 }
22 inline void pushup(int x) {
23     siz[x] = siz[ch[x][0]] + siz[ch[x][1]] + 1;
24 }
25 inline int makenode(int x) {
26     ++tn;val[tn] = x;siz[tn] = 1;key[tn] = rand();return tn;
27 }
28 
29 int Merge(int x,int y) {
30     if (!x || !y) return x + y;
31     if (key[x] < key[y]) {
32         ch[x][1] = Merge(ch[x][1],y);
33         pushup(x); return x;    
34     }
35     else {
36         ch[y][0] = Merge(x,ch[y][0]);
37         pushup(y); return y;
38     }
39 }
40 void Split(int now,int k,int &x,int &y) {
41     if (!now) x = y = 0;
42     else {
43         if (val[now] <= k) 
44             x = now,Split(ch[now][1],k,ch[now][1],y);
45         else 
46             y = now,Split(ch[now][0],k,x,ch[now][0]);
47         pushup(now);
48     }
49 }
50 inline int getkth(int p,int k) {
51     while (true) {
52         if (k == siz[ch[p][0]] + 1) return p;
53         if (ch[p][0] && k <= siz[ch[p][0]]) p = ch[p][0];
54         else k-= ((ch[p][0] ? siz[ch[p][0]] : 0) + 1),p = ch[p][1];
55     }
56 }
57 int main() {
58     int x,y,z,opt,k,n = read();
59     while (n--) {
60         opt = read(),k = read();
61         if (opt==1) {
62             Split(Root,k,x,y);
63             Root = Merge(Merge(x,makenode(k)),y);
64         }
65         else if (opt==2) {
66             Split(Root,k,x,y);
67             Split(x,k-1,x,z);
68             z = Merge(ch[z][0],ch[z][1]);
69             Root = Merge(Merge(x,z),y);
70         }
71         else if (opt==3) {
72             Split(Root,k-1,x,y);
73             printf("%d\n",siz[x]+1);
74             Root = Merge(x,y);
75         }
76         else if (opt==4) 
77             printf("%d\n",val[getkth(Root,k)]);
78         else if (opt==5) {
79             Split(Root,k-1,x,y);
80             printf("%d\n",val[getkth(x,siz[x])]);
81             Root = Merge(x,y);
82         }
83         else {
84             Split(Root,k,x,y);
85             printf("%d\n",val[getkth(y,1)]);
86             Root = Merge(x,y);
87         }
88     }
89     return 0;
90 }