SPOJ 3273 - Order statistic set Treap模板题
题意:
有个集合..起初为空...有Q(1<=Q<=200000)个操作或者询问
操作insert: I x: 若集合中没有x...把x放入
操作delete: D x:若集合中有x..删去x
询问K-th: K x:输出集合中第x大的数是什么...若总个数没有k个输出invalid
询问Count: C x:输出集合中有多少个数比x小...
题解:
裸treap了....操作还挺全的...为了找到第k大的数以及小于某个数的个数...每个点上加一个变量.用于存储以其为根的子树有多少个节点...insert和delete参考了nocow中CmYkRgB123的代码...两个询问时原创的~不用递归相当和谐..
Program:
#include<iostream>
#include<stdio.h>
#include<string.h>
#include<cmath>
#include<queue>
#include<stack>
#include<set>
#include<time.h>
#include<map>
#include<algorithm>
#define ll long long
#define eps 1e-5
#define oo 1000000007
#define pi acos(-1.0)
#define MAXN 200005
using namespace std;
struct node
{
int l,r,key,fix,size;
};
struct treap
{
node h[MAXN];
int root,num;
void initial() { srand((int)time(0)),num=root=0; }
void rot_l(int &x)
{
int R=h[x].r,L=h[x].l;
h[x].size=h[x].size-h[R].size+h[h[R].l].size;
h[R].size+=h[L].size+1;
h[x].r=h[R].l,h[R].l=x;
x=R;
}
void rot_r(int &x)
{
int L=h[x].l,R=h[x].r;
h[x].size=h[x].size-h[L].size+h[h[L].r].size;
h[L].size+=h[R].size+1;
h[x].l=h[L].r,h[L].r=x;
x=L;
}
bool insert(int &k,int key)
{
if (!k)
{
k=++num;
h[k].l=h[k].r=0,h[k].size=1;
h[k].key=key,h[k].fix=rand();
return true;
}
if (h[k].key==key) return false;
if (h[k].key>key)
{
if (!insert(h[k].l,key)) return false;
h[k].size++;
if (h[h[k].l].fix>h[k].fix) rot_r(k);
return true;
}else
{
if (!insert(h[k].r,key)) return false;
h[k].size++;
if (h[h[k].r].fix>h[k].fix) rot_l(k);
return true;
}
}
int count(int key)
{
int g=0,k=root;
while (k)
{
if (h[k].key>key) k=h[k].l;
else g+=h[h[k].l].size+1,k=h[k].r;
}
return g;
}
int k_th(int kth)
{
int g=0,k=root;
if (h[root].size<kth) return -1;
while (h[h[k].l].size+g+1!=kth)
{
if (h[h[k].l].size+g+1>=kth) k=h[k].l;
else g+=h[h[k].l].size+1,k=h[k].r;
}
return h[k].key;
}
bool del(int &k,int key)
{
if (!k) return false;
if (h[k].key>key)
{
if (!del(h[k].l,key)) return false;
h[k].size--;
}
else
if (h[k].key<key)
{
if (!del(h[k].r,key)) return false;
h[k].size--;
}
else
{
if (!h[k].l && !h[k].r) k=0;
else
if (!h[k].l) k=h[k].r;
else
if (!h[k].r) k=h[k].l;
else
{
if (h[h[k].l].fix<h[h[k].r].fix)
{
rot_l(k);
if (!del(h[k].l,key)) return false;
h[k].size--;
}
else
{
rot_r(k);
if (!del(h[k].r,key)) return false;
h[k].size--;
}
}
}
return true;
}
}mytreap;
int main()
{
int m,x;
char c;
mytreap.initial();
scanf("%d",&m);
while (m--)
{
do { c=getchar(); } while (c<'A' || c>'Z');
scanf("%d",&x);
if (c=='I') mytreap.insert(mytreap.root,x); else
if (c=='C') printf("%d\n",mytreap.count(x-1)); else
if (c=='K')
{
x=mytreap.k_th(x);
if (x==-1) printf("invalid\n");
else printf("%d\n",x);
}else
if (c=='D') mytreap.del(mytreap.root,x);
}
return 0;
}