P3369 【模板】普通平衡树FHQtreap

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

 #include<cstdio>
 #include<algorithm>

 using namespace std;

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

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

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