UESTC 395 Dynamic Query System --Treap
题意:让你维护一个集合,有8种操作:
1. I x 插入一个数
2. R x 删除x
3. S 输出总的数个数(集合大小)
4. L x 查询小于x的数的个数
5. W k 查询集合中数从小到大排列的第k个数
6. C x 查询x的个数
7. MI 查询集合中最小的数
8. MA 查询集合中最大的数
一道平衡树模拟的裸题,直接套版然后改下细节就行了。
代码:
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
#include <iostream> #include <cstdio> #include <cstring> #include <cstdlib> #include <algorithm> using namespace std; struct node { node *ch[2]; int v,r,sz,cnt; void UP(){ sz = ch[0]->sz + ch[1]->sz + cnt; } int cmp(int x)const { if(x == v) return -1; return x > v; } }*null,*root; void create(node *&o,int v=0) { o = new node(); o->sz = o->cnt = 1; o->v = v; o->r = rand(); o->ch[0] = o->ch[1] = null; } void rotate(node *&o,int d) { node *p = o->ch[d]; o->ch[d] = p->ch[d^1]; p->ch[d^1] = o; o->UP(),p->UP(); o = p; } int countx(node *o,int v) { while(o != null) { int d = o->cmp(v); if(d == -1) return o->cnt; o = o->ch[d]; } return 0; } void insert(node *&o,int v) { if(o == null) { create(o,v); return; } int d = o->cmp(v); if(d == -1) { o->cnt++; o->UP(); return; } insert(o->ch[d],v); if(o->r > o->ch[d]->r) rotate(o,d); if(o != null) o->UP(); } void del(node *&o,int v) { if(o == null) return; int d = o->cmp(v); if(d == -1) { if(o->cnt > 1) o->cnt--; else { if(o->ch[0] == null) o = o->ch[1]; else if(o->ch[1] == null) o = o->ch[0]; else { int d2 = o->ch[1]->r < o->ch[0]->r; if(o->ch[d2] == null) { delete o; o = null; return; } rotate(o,d2); del(o->ch[d2^1],v); } } } else del(o->ch[d],v); if(o != null) o->UP(); } int Kth(node *o,int k) { if(o == null || k == 0 || k > o->sz) return -1; int left = o->ch[0]->sz; if(k > left && k <= left + o->cnt) return o->v; if(k > left + o->cnt) return Kth(o->ch[1],k-left-o->cnt); return Kth(o->ch[0],k); } int Count(node *o,int v) { if(o == null) return 0; int left = o->ch[0]->sz; if(o->v == v) return left; else if(v < o->v) return Count(o->ch[0],v); return Count(o->ch[1],v)+left+o->cnt; } void init() { create(null); null->sz = 0; root = null; } int main() { int t,n,i,x; char ss[3]; scanf("%d",&t); while(t--) { init(); scanf("%d",&n); while(n--) { scanf("%s",ss); if(ss[0] == 'M') { if(ss[1] == 'I') printf("%d\n",Kth(root,1)); else printf("%d\n",Kth(root,root->sz)); } else if(ss[0] == 'I') { scanf("%d",&x); insert(root,x); } else if(ss[0] == 'S') { printf("%d\n",root->sz); } else if(ss[0] == 'R') { scanf("%d",&x); del(root,x); } else if(ss[0] == 'L') { scanf("%d",&x); printf("%d\n",Count(root,x)); } else if(ss[0] == 'W') { scanf("%d",&x); printf("%d\n",Kth(root,x)); } else if(ss[0] == 'C') { scanf("%d",&x); printf("%d\n",countx(root,x)); } } } return 0; }
作者:whatbeg
出处1:http://whatbeg.com/
出处2:http://www.cnblogs.com/whatbeg/
本文版权归作者和博客园共有,欢迎转载,但未经作者同意必须保留此段声明,且在文章页面明显位置给出原文连接,否则保留追究法律责任的权利。
更多精彩文章抢先看?详见我的独立博客: whatbeg.com