巴蜀1322 第k小的数

Description

　　现在已有N个整数，你有以下三种操作：

1 A：表示加入一个值为A的整数；

2 B：表示删除其中值为B的整数；

3 K：表示输出这些整数中第K小的数；

Input

第一行，两个整数N，M，表示最开始有N个整数，总共有M个操作

第二行用空格隔开的N个整数

接下来M行，每行表示一个操作

Output

若干行，一行一个整数，表示所求的第K小的数字

平衡树模板题。

#include<cstdio>
#include<cstdlib>
struct node
{
int size,x,v,k[2];
}t[3100000];
int root,size;
void up(int p)
{
t[p].size=t[t[p].k[0]].size+t[t[p].k[1]].size+1;
}
void rot(int &p,bool b)
{
int x=t[p].k[b];
int y=t[x].k[!b];
t[p].k[b]=y;
t[x].k[!b]=p;
up(p);
up(x);
p=x;
}
void ins(int &p,int x)
{
if (p==0)
{
p=++size;
t[p].x=x;
t[p].v=rand();
t[p].size=1;
return;
}
if (t[p].x==x) return;
bool b=x>t[p].x;
ins(t[p].k[b],x);
up(p);
if (t[t[p].k[b]].v>t[p].v)
rot(p,b);
}
void del(int &p,int x)
{
if (p==0) return;
if (t[p].x==x)
{
if (t[p].k[0]==0&&t[p].k[1]==0)
{
p=0;
return;
}
bool b=t[t[p].k[0]].v<t[t[p].k[1]].v;
rot(p,b);
del(t[p].k[!b],x);
up(p);
}
del(t[p].k[x>t[p].x],x);
up(p);
}
int find(int p,int x)
{
if (p==0) return 0;
if (t[t[p].k[0]].size>=x) return find(t[p].k[0],x);
if (t[t[p].k[0]].size+1==x) return t[p].x;
return find(t[p].k[1],x-t[t[p].k[0]].size-1);
}
int main()
{
int m,n,i,j,k,x,y,z;
scanf("%d%d",&n,&m);
for (i=1;i<=n;i++)
{
scanf("%d",&x);
ins(root,x);
}
for (i=1;i<=m;i++)
{
scanf("%d%d",&x,&y);
if (x==1) ins(root,y);
if (x==2) del(root,y);
if (x==3) printf("%d\n",find(root,y));
}
}