[bzoj3224][tyvj1728][普通平衡树] (pb_ds库自带红黑树)

Description

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

1. 插入x数

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

3. 查询x数的排名(若有多个相同的数，因输出最小的排名)

4. 查询排名为x的数

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

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

Input

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

Output

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

Sample Input

10

1 106465

4 1

1 317721

1 460929

1 644985

1 84185

1 89851

6 81968

1 492737

5 493598

Sample Output

106465

84185

492737

HINT

1.n的数据范围：n<=100000

2.每个数的数据范围：[-1e7,1e7]

数据如下http://pan.baidu.com/s/1jHMJwO2

pb_ds中红黑树500ms水过

#include <bits/stdc++.h>
#include <ext/pb_ds/tree_policy.hpp>
#include <ext/pb_ds/assoc_container.hpp>
using namespace std;
using namespace __gnu_pbds;
struct pt{
int first,second;
pt(int x,int y) :first(x),second(y) {}
bool operator<(const pt h)const{
return first<h.first || (first==h.first&&second<h.second);
}
bool operator==(const pt h)const{
return first==h.first&&second==h.second;
}
};
typedef tree<pt,null_type,less< pt >,rb_tree_tag,tree_order_statistics_node_update> rbtree;
inline char Blue(){
static char B[1<<9],*S=B,*T=B;
if(S==T)T=(S=B)+fread(B,1,1<<9,stdin);
return *S++;
}
inline int Rin(){
int x=0,c=Blue(),f=1;
for(;c<48||c>57;c=Blue())
if(!(c^45))f=-1;
for(;c>47&&c<58;c=Blue())
x=(x<<1)+(x<<3)+c-48;
return x*f;
}
int T,opt,x;
map<int,int>s;
rbtree t;
int main(){
T=Rin();
while(T--){
opt=Rin(),x=Rin();
switch(opt){
case 1:t.insert(pt(x,s[x]++));break;
case 2:t.erase(pt(x,--s[x]));break;
case 3:
printf("%d\n",t.order_of_key(pt(x,0))+1);
break;
case 4:
printf("%d\n",t.find_by_order(x-1)->first);
break;
case 5:
printf("%d\n",t.find_by_order(t.order_of_key(pt(x,0))-1)->first);
break;
case 6:
printf("%d\n",t.find_by_order(t.order_of_key(pt(x,s[x]-1))+(t.find(pt(x,0))==t.end()?0:1))->first);
break;
default:break;
}
}
return 0;
}