【bzoj 3224】代码:
#include<algorithm>
#include<iostream>
#include<cstring>
#include<string>
#include<cstdio>
#include<cmath>
#include<cstdlib>
using namespace std;
const int inf=2e9+7;
struct node{
int x,num,siz;
node *ch[2],*fa;
node(int);
void maintain(){siz=ch[0]->siz+ch[1]->siz+num;}
int son(){
if(fa->ch[0]==this) return 0;
if(fa->ch[1]==this) return 1;
return -1;
}
}*null=new node(0);
node:: node(int _){
x=_;num=siz=null ? 1 : 0;
fa=ch[0]=ch[1]=null ? null : this;
}
struct Tree{
node *root,*tmp;
void ST(){
root=new node(-inf);
root->ch[1]=new node(inf);
root->ch[1]->fa=root;
root->maintain();
}
node* insert(node* &p,int u){
if(p==null){p=new node(u);return p;}
node* t;
if(p->x==u){p->num++;p->siz++;t=p;}
else if(p->x<u) {t=insert(p->ch[1],u);p->ch[1]->fa=p;}
else {t=insert(p->ch[0],u);p->ch[0]->fa=p;}
p->maintain();return t;
}
void Rotate(node* p,bool f){
node* t=p->ch[f^1];
p->ch[f^1]=t->ch[f];
if(t->ch[f]!=null) t->ch[f]->fa=p;
t->ch[f]=p;
p->maintain();t->maintain();
if(~p->son()) p->fa->ch[p->son()]=t;
t->fa=p->fa;p->fa=t;
}
void splay(node* p,bool f){
if(f){
while(~p->son()){
int dir=p->son();
if(p->fa->son()==dir) Rotate(p->fa->fa,dir^1);
Rotate(p->fa,dir^1);
}
root=p;
}
else {
while(~p->son()){
int dir=p->son();
if(p->fa==root) return ;
if(p->fa->fa==root) {Rotate(p->fa,dir^1);return ;}
if(p->fa->son()==dir) Rotate(p->fa->fa,dir^1);
Rotate(p->fa,dir^1);
}
}
}
node* ch(node* p,int u){
if(p->x==u) return p;
if(p->x<u) return ch(p->ch[1],u);
if(p->x>u) return ch(p->ch[0],u);
}
void ch0(node* p,int u){
if(p==null) return ;
if(p->x<u) {tmp=p;ch0(p->ch[1],u);}
else ch0(p->ch[0],u);
}
void ch1(node* p,int u){
if(p==null) return ;
if(p->x>u) {tmp=p;ch1(p->ch[0],u);}
else ch1(p->ch[1],u);
}
int search(node* p,int sum){
if(p->ch[0]->siz>=sum) return search(p->ch[0],sum);
else if(p->ch[0]->siz+p->num>=sum) return p->x;
else return search(p->ch[1],sum-p->ch[0]->siz-p->num);
}
void solve(int u,int num){
tmp=null;
if(num==1){
tmp=ch(root,u);splay(tmp,1);
printf("%d\n",root->ch[0]->siz);
}
if(num==3){
ch0(root,u);splay(tmp,1);
printf("%d\n",root->x);
}
if(num==4){
ch1(root,u);splay(tmp,1);
printf("%d\n",root->x);
}
if(num==2) printf("%d\n",search(root,u+1));
}
void Insert(int u){
node* t=insert(root,u);
splay(t,1);
}
void Del(int u){
node *r1,*r2;
tmp=null;ch0(root,u);r1=tmp;
tmp=null;ch1(root,u);r2=tmp;
splay(r1,1);splay(r2,0);
if(root->ch[1]->ch[0]->num>=2) {root->ch[1]->ch[0]->siz--;root->ch[1]->ch[0]->num--;}
else {delete(root->ch[1]->ch[0]);root->ch[1]->ch[0]=null;}
root->ch[1]->maintain();root->maintain();
}
}T;
int n,ai,bi;
int main(){
T.ST();
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%d%d",&ai,&bi);
if(ai==1) T.Insert(bi);//插入x
if(ai==2) T.Del(bi);//删除x
if(ai==3) T.solve(bi,1);//查询x数的排名
if(ai==4) T.solve(bi,2);//查询排名为x的数
if(ai==5) T.solve(bi,3);//求x的前驱
if(ai==6) T.solve(bi,4);//求x的后继
}
return 0;
}