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

 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 }

P3369 【模板】普通平衡樹FHQtreap