P3369 【模板】普通平衡树(Treap/SBT)
题目描述
您需要写一种数据结构(可参考题目标题),来维护一些数,其中需要提供以下操作:
-
插入x数
-
删除x数(若有多个相同的数,因只删除一个)
-
查询x数的排名(排名定义为比当前数小的数的个数+1。若有多个相同的数,因输出最小的排名)
-
查询排名为x的数
-
求x的前驱(前驱定义为小于x,且最大的数)
- 求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 ",siz[x]+1); 74 Root = Merge(x,y); 75 } 76 else if (opt==4) 77 printf("%d ",val[getkth(Root,k)]); 78 else if (opt==5) { 79 Split(Root,k-1,x,y); 80 printf("%d ",val[getkth(x,siz[x])]); 81 Root = Merge(x,y); 82 } 83 else { 84 Split(Root,k,x,y); 85 printf("%d ",val[getkth(y,1)]); 86 Root = Merge(x,y); 87 } 88 } 89 return 0; 90 }