BZOJ 3224 TYVJ 1728 普通平衡树 [Treap树模板]
2016-06-05 18:48
337 查看
3224: Tyvj 1728 普通平衡树
Time Limit: 10 Sec Memory Limit: 128 MB
Submit: 7390 Solved: 3122
[Submit][Status][Discuss]
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
101 106465
4 1
1 317721
1 460929
1 644985
1 84185
1 89851
6 81968
1 492737
5 493598
Sample Output
10646584185
492737
HINT
1.n的数据范围:n<=1000002.每个数的数据范围:[-1e7,1e7]
数据如下http://pan.baidu.com/s/1jHMJwO2 treap的模板题,涉及了treap的各种操作 操作函数中运用了传引用的方式改变上一层结点的子节点值,很方便
/*by SilverN*/ #include<iostream> #include<cstdio> #include<cmath> #include<cstring> #include<algorithm> using namespace std; int n; struct treap{ int l,r;//左右子树 int val,ct;//值 重复次数 int size,rand;//管控数量,随机权值 }t[120000]; int root=0,cnt=0; int ans; void update(int k){ t[k].size=t[t[k].l].size+t[t[k].r].size+t[k].ct; } void lt(int &k){//左旋 int now=t[k].r; t[k].r=t[now].l; t[now].l=k; t[now].size=t[k].size; update(k); k=now; return; } void rt(int &k){ int now=t[k].l; t[k].l=t[now].r; t[now].r=k; t[now].size=t[k].size; update(k); k=now; return; } void insert(int &k,int x){//&k是为了建新节点 x为插入值 if(k==0){//建新节点 t[++cnt].val=x; t[cnt].size=1; t[cnt].ct=1; t[cnt].rand=rand(); k=cnt; return; } t[k].size++; if(t[k].val==x) t[k].ct++;//与节点等值 else if(x>t[k].val){ insert(t[k].r,x); if(t[t[k].r].rand<t[k].rand) lt(k); }else{//x<t[k].val insert(t[k].l,x); if(t[t[k].l].rand<t[k].rand) rt(k); } return; } void del(int &k,int x){ if(k==0)return; if(t[k].val==x){ if(t[k].ct>1){t[k].ct--;t[k].size--;return;} if(t[k].l*t[k].r==0)k=t[k].l+t[k].r;//如果k是链结点(只有一个子节点),由其子节点补位 else{ if(t[t[k].l].rand<t[t[k].r].rand){ rt(k); del(k,x); } else{ lt(k); del(k,x); } } return; } t[k].size--; if(x>t[k].val)del(t[k].r,x); if(x<t[k].val)del(t[k].l,x); return; } void ask_p(int k,int x,int mode){//前驱 //mode==1 ->前驱 mode==2 ->后驱 if(mode==1){ if(!k)return; if(t[k].val<x){//依照二叉树的性质,要找小的往左查,要找大的往右查 ans=t[k].val; ask_p(t[k].r,x,1); }else ask_p(t[k].l,x,1); return; } if(mode==2){ if(!k)return; if(t[k].val>x){ ans=t[k].val; ask_p(t[k].l,x,2); }else ask_p(t[k].r,x,2); } return; } int ask_rank(int k,int x){//已知数字问排名 if(!k)return 0; if(t[k].val==x)return t[t[k].l].size+1; if(t[k].val<x)return t[k].ct+t[t[k].l].size+ask_rank(t[k].r,x); else return ask_rank(t[k].l,x); } int ask_num(int k,int x){// 已知排名问数字 if(!k)return 0; if(x<=t[t[k].l].size)return ask_num(t[k].l,x);//排名小于左子树包含结点数量,则往左查 if(x>t[t[k].l].size+t[k].ct)return ask_num(t[k].r,x-t[t[k].l].size-t[k].ct); //排名大于“左子树结点数加父结点重复数”,则往右查 else return t[k].val;//否则返回父结点值 } int main(){ int opt,x; scanf("%d",&n); while(n--){ scanf("%d%d",&opt,&x); switch(opt){ case 1: insert(root,x); break; case 2: del(root,x); break; case 3: printf("%d\n",ask_rank(root,x)); break; case 4: printf("%d\n",ask_num(root,x)); break; case 5: ask_p(root,x,1);printf("%d\n",ans);break;//前驱 case 6: ask_p(root,x,2);printf("%d\n",ans);break;//前驱 } // for(int i=1;i<=cnt;i++)printf("%d ",t[i].val); // cout<<endl; } return 0; }
相关文章推荐
- bzoj 2054: 疯狂的馒头
- java 开发界面利器WindowBuilder Pro 安装方法
- 开源爬虫OpenSpider(一)----------------第一次提交
- Linux netstat命令详解
- 【leetcode】141/142Linked List Cycle(Floyd判圈算法)
- 第七次c++作业
- 抽象类
- 第二章:介绍 Olly Debug
- 静态变量和成员变量有什么区别?应该怎么调用?
- C++第七次上机实验—数组素数排序
- SSH框架介绍以及如何搭建(myEclipse下)
- Unity3d优化
- c++作业7
- HtmlParser基础教程 - 在java中解析获取的HttpResponse
- 第十二周项目一(3)-实现复数类中的运算符重载
- 视频转换成flv注意事项
- 三层架构
- HTTP协议学习# (一)request 和response 解析
- 二维数组中的查找
- iOS开发中几种常用设计模式