想必大家都知道一种叫做二叉搜索树这东西吧,那么我们知道,在某些特殊情况下,二叉搜索树会退化成一条链,而且如果出题人成心想卡你的话也很简单,分分钟把你(n log n)的期望卡成。那么我们该如何避免这种尴尬状况的出现呢?我们的范浩强大佬就创造了一种名为Treap的算法。

那么这个算法是如何实现的呢?

首先,我们发现:

当我们将一组数放入一棵二叉搜索树的顺序改变时,那么你最终得到的二叉搜索树也会发生变化。

这便是Treap算法优化二叉搜索树的突破口。这二叉搜索树的这一个特点也就意味着我们完全可以将一棵已经退化成一条链的二叉搜索树优化成一个两边相对平衡的二叉搜索树。

在这里,我隆重介绍一种在一颗二叉搜索树中可谓BUG级别的操作——左/右旋。

这个操作可以在满足一颗二叉搜索树性质的同时交换父亲与儿子的位置,也就是说这个操作可以逆天到让爸爸给他儿子当儿子,儿子当爸爸,而且还是被法律所允许的!!!(想想就恐(ji)(dong))嗯,具体操作如下图:

a,b,c分别为三个节点,其中我们要交换的是b节点和c节点。t为a的整颗左子树,x为b的整颗左子树,y为c的整颗左子树,z为c的整颗右子树。可以看见,右图由左图左旋得到,而左图就是由右图右旋得到的。

学会了这个操作也就意味这我们可以对这颗二叉搜索树为所欲为了!

所以我们现在所要考虑的就是如何做到让其相对平衡?

其实答案很简单,就是给每个数随机一个第二关键字,并且在满足一颗二叉搜索树性质的同时,用来维护这个第二关键字(我这里采用了大根堆),然后我们便能够得到一颗相对平衡的二叉搜索树了!

这也便是Treap这个名字的由来:Tree(树)+Heap(堆)

1、废话不多说,先给大家附上我的变量定义

type
treap=record
key,fix,cnt,siz:longint;
ch:array[0..1]of longint;

//ch数组表示它连接的儿子节点ch[-1]为左儿子,ch[1]为右儿子(这么定义就是为了偷懒。。。)
//key为第一关键字(就是它的数值)fix为第二关键字(我们随机出来的)
//cnt表示与当前节点数值相同的点的个数,size表示以他为根的子树的大小
//father为它的父亲节点
//注意,因为旋转操作是涉及到祖孙三代的,所以既要记录儿子又要记录父亲。

end; 

2、逆天的旋转操作

procedure rotate(var x:longint;d:longint);
var
son:longint;
begin
son:=t[x].ch[d];
t[x].ch[d]:=t[son].ch[1-d];
t[son].ch[1-d]:=x;
update(x); update(son);
x:=son;
end;

3、更新操作

procedure update(x:longint);
begin
t[x].siz:=t[x].cnt+t[t[x].ch[0]].siz+t[t[x].ch[1]].siz;
end;

4、插入操作

删除操作分为三步:

1、找到该数在Treap中的位置,并将路径上的点size域+1;

2、插入该数,若树中已有该数,则将该数的cnt域+1,否则就新建一个叶子节点;

3、回溯时,判断儿子和父亲的fix(第二关键字)域是否满足堆的性质,若不满足则用旋转操作把父亲结点旋下来,将儿子节点转上去,使其继续满足堆的性质;

procedure inset(var k:longint;key:longint);
var
d:longint;
begin
if k=0 then
begin
inc(cnt); k:=cnt; //新建节点
t[k].key:=key; t[k].fix:=random(100000000);
t[k].cnt:=1; t[k].siz:=1;
exit;
end else inc(t[k].siz); //路径上经过的siz都加1
if t[k].key=key then begin inc(t[k].cnt); exit; end;
d:=ord(key>t[k].key); //往哪走
inset(t[k].ch[d],key);
if t[t[k].ch[d]].fix>t[k].fix then rotate(k,d); //维护堆的性质
end;

5、删除操作

找到删除结点的位置后有三种情况:

1、该节点有多个,也就是t[x].cnt>1:此时只要将t[x].cnt-1即可。

2、它只有一个儿子:用它的儿子来代替它。(没儿子直接删)

3、它有两个儿子:将它第二关键字值较大(若是小根堆则为较小)的节点旋上来,继续递归,直至到达1或2。

procedure delet(var k:longint;key:longint);
begin
if k=0 then exit; //找不到
if t[k].key=key then
begin
if t[k].cnt>1 then //情况1
begin dec(t[k].cnt); dec(t[k].siz); exit; end;
if (t[k].ch[0]=0)or(t[k].ch[1]=0) then //情况2
begin k:=t[k].ch[0]+t[k].ch[1]; exit; end;
rotate(k,ord(t[t[k].ch[0]].fix<t[t[k].ch[1]].fix)); //情况3
delet(k,key);
end else
begin
dec(t[k].siz); //路径上的节点siz都减1
delet(t[k].ch[ord(t[k].key<key)],key); //往哪走
end;
end;

6、查排名

function rank(k,key:longint):longint;
begin
if k=0 then exit(0); //找不到
if key=t[k].key then exit(t[t[k].ch[0]].siz) //找到了,这时整棵左子树都比它小
else if key<t[k].key then exit(rank(t[k].ch[0],key)) //往左走
else exit(t[t[k].ch[0]].siz+rank(t[k].ch[1],key)+t[k].cnt); //往右走,这时整棵左子树和节点都比它小
end;

7、查找第k小数

感觉和上面查排名的操作差不多,就是反了一下罢了。(这里的k是节点编号,其实找的是第x小数)

和权值线段树上二分很像哦。

function find_kth(k,x:longint):longint;
begin
if k=0 then exit(-1); //找不到
if x<=t[t[k].ch[0]].siz then exit(find_kth(t[k].ch[0],x)); //左子树规模比x大,说明在左子树中
if x<=t[t[k].ch[0]].siz+t[k].cnt then exit(t[k].key); //找到了
exit(find_kth(t[k].ch[1],x-t[k].cnt-t[t[k].ch[0]].siz)); //在右子树中
end;

8、找前驱/后继

前驱就是在它的左子树中找一个最大的数,而后继则为在它的右子树中找一个最小的数。

在注意一下在找最大/最小值时别忘了它的父亲就行。(因为可能和它的值一样)

function find_pre(k,key:longint):longint;
begin
if k=0 then exit(-maxlongint);
if t[k].key>=key then exit(find_pre(t[k].ch[0],key));
exit(max(find_pre(t[k].ch[1],key),t[k].key));
end;
function find_suc(k,key:longint):longint;
begin
if k=0 then exit(maxlongint);
if t[k].key<=key then exit(find_suc(t[k].ch[1],key));
exit(min(find_suc(t[k].ch[0],key),t[k].key));
end;

下面附上我完整的代码:

type
treap=record
key,fix,cnt,siz:longint;
ch:array[0..1]of longint;
end;
var
t:array[-1..100000]of treap;
i,opt,x,cnt,q,root:longint;
function min(a,b:longint):longint;
begin
if a<b then exit(a) else exit(b);
end;
function max(a,b:longint):longint;
begin
if a>b then exit(a) else exit(b);
end;
procedure update(x:longint);
begin
t[x].siz:=t[x].cnt+t[t[x].ch[0]].siz+t[t[x].ch[1]].siz;
end;
procedure rotate(var x:longint;d:longint);
var
son:longint;
begin
son:=t[x].ch[d];
t[x].ch[d]:=t[son].ch[1-d];
t[son].ch[1-d]:=x;
update(x); update(son);
x:=son;
end;
procedure inset(var k:longint;key:longint);
var
d:longint;
begin
if k=0 then
begin
inc(cnt); k:=cnt;
t[k].key:=key; t[k].fix:=random(100000000);
t[k].cnt:=1; t[k].siz:=1;
exit;
end else inc(t[k].siz);
if t[k].key=key then begin inc(t[k].cnt); exit; end;
d:=ord(key>t[k].key);
inset(t[k].ch[d],key);
if t[t[k].ch[d]].fix>t[k].fix then rotate(k,d);
end;
procedure delet(var k:longint;key:longint);
begin
if k=0 then exit;
if t[k].key=key then
begin
if t[k].cnt>1 then
begin dec(t[k].cnt); dec(t[k].siz); exit; end;
if (t[k].ch[0]=0)or(t[k].ch[1]=0) then
begin k:=t[k].ch[0]+t[k].ch[1]; exit; end;
rotate(k,ord(t[t[k].ch[0]].fix<t[t[k].ch[1]].fix));
delet(k,key);
end else
begin
dec(t[k].siz);
delet(t[k].ch[ord(t[k].key<key)],key);
end;
end;
function rank(k,key:longint):longint;
begin
if k=0 then exit(0);
if key=t[k].key then exit(t[t[k].ch[0]].siz)
else if key<t[k].key then exit(rank(t[k].ch[0],key))
else exit(t[t[k].ch[0]].siz+rank(t[k].ch[1],key)+t[k].cnt);
end;
function find_kth(k,x:longint):longint;
begin
if k=0 then exit(-1);
if x<=t[t[k].ch[0]].siz then exit(find_kth(t[k].ch[0],x));
if x<=t[t[k].ch[0]].siz+t[k].cnt then exit(t[k].key);
exit(find_kth(t[k].ch[1],x-t[k].cnt-t[t[k].ch[0]].siz));
end;
function find_pre(k,key:longint):longint;
begin
if k=0 then exit(-maxlongint);
if t[k].key>=key then exit(find_pre(t[k].ch[0],key));
exit(max(find_pre(t[k].ch[1],key),t[k].key));
end;
function find_suc(k,key:longint):longint;
begin
if k=0 then exit(maxlongint);
if t[k].key<=key then exit(find_suc(t[k].ch[1],key));
exit(min(find_suc(t[k].ch[0],key),t[k].key));
end;
begin
randomize;
read(q);
root:=0;
while q>0 do
begin
read(opt,x);
if opt=1 then inset(root,x) else
if opt=2 then delet(root,x) else
if opt=3 then writeln(rank(root,x)+1) else
if opt=4 then writeln(find_kth(root,x)) else
if opt=5 then writeln(find_pre(root,x)) else
if opt=6 then writeln(find_suc(root,x));
dec(q);
end;
end.

[学习笔记] Treap的更多相关文章

  1. 学习笔记 | treap | splay

    目录 前言 treap 它的基本操作 前言 不会数据结构选手深深地感受到了来自treap的恶意QwQ 在听的时候感觉自己听得听懂的??大概只是听懂了它的意思 代码是怎么写都感觉写不好╮(╯﹏╰)╭ 菜 ...

  2. 平衡树学习笔记(2)-------Treap

    Treap 上一篇:平衡树学习笔记(1)-------简介 Treap是一个玄学的平衡树 为什么说它玄学呢? 还记得上一节说过每个平衡树都有自己的平衡方式吗? 没错,它平衡的方式是......rand ...

  3. 左偏树 / 非旋转treap学习笔记

    背景 非旋转treap真的好久没有用过了... 左偏树由于之前学的时候没有写学习笔记, 学得也并不牢固. 所以打算写这么一篇学习笔记, 讲讲左偏树和非旋转treap. 左偏树 定义 左偏树(Lefti ...

  4. 「学习笔记」Treap

    「学习笔记」Treap 前言 什么是 Treap ? 二叉搜索树 (Binary Search Tree/Binary Sort Tree/BST) 基础定义 查找元素 插入元素 删除元素 查找后继 ...

  5. Treap-平衡树学习笔记

    平衡树-Treap学习笔记 最近刚学了Treap 发现这种数据结构真的是--妙啊妙啊~~ 咳咳.... 所以发一发博客,也是为了加深蒟蒻自己的理解 顺便帮助一下各位小伙伴们 切入正题 Treap的结构 ...

  6. [学习笔记]平衡树(Splay)——旋转的灵魂舞蹈家

    1.简介 首先要知道什么是二叉查找树. 这是一棵二叉树,每个节点最多有一个左儿子,一个右儿子. 它能支持查找功能. 具体来说,每个儿子有一个权值,保证一个节点的左儿子权值小于这个节点,右儿子权值大于这 ...

  7. OI知识点|NOIP考点|省选考点|教程与学习笔记合集

    点亮技能树行动-- 本篇blog按照分类将网上写的OI知识点归纳了一下,然后会附上蒟蒻我的学习笔记或者是我认为写的不错的专题博客qwqwqwq(好吧,其实已经咕咕咕了...) 基础算法 贪心 枚举 分 ...

  8. 平衡树学习笔记(6)-------RBT

    RBT 上一篇:平衡树学习笔记(5)-------SBT RBT是...是一棵恐怖的树 有多恐怖? 平衡树中最快的♂ 不到200ms的优势,连权值线段树都无法匹敌 但是,通过大量百度,发现RBT的代码 ...

  9. 平衡树学习笔记(5)-------SBT

    SBT 上一篇:平衡树学习笔记(4)-------替罪羊树 所谓SBT,就是Size Balanced Tree 它的速度很快,完全碾爆Treap,Splay等平衡树,而且代码简洁易懂 尤其是插入节点 ...

随机推荐

  1. Unity代码混淆

    https://www.zhihu.com/question/25414422 http://blog.csdn.net/kun1234567/article/details/7917847 http ...

  2. Infinite Inversions(树状数组+离散化)

    思路及代码参考:https://blog.csdn.net/u014800748/article/details/45420085 There is an infinite sequence cons ...

  3. vue 页面首次加载缓慢原因及解决方案

    第一次打包vue的项目部署到服务器,发现首次加载特别的缓慢要几十秒才加载出来,完全没有在本地开发环境上那么流畅. 主要原因是页面在打包后没有进行相关的配置导致资源文件特别大,一次想要全部加载完成回特别 ...

  4. 为什么要做一款ERP软件——开源软件诞生7

    技术之外的探讨--第7篇 用日志记录“开源软件”的诞生 赤龙ERP开源地址: 点亮星标,感谢支持,与开发者交流 kzca2000 码云:https://gitee.com/redragon/redra ...

  5. Activiti7 网关(并行网关)

    什么是并行网关? 并行网关允许将流程分成多条分支,也可以将多条分支合并到一起,并行网关是基于进入和外出顺序流的 fork分支: 并行后的所有外出顺序流,为每个顺序流都创建一个并发分支 jion汇聚: ...

  6. Oracle数据库之表与表数据操作

    一.SQL语言 SQL语言分为四种,分别是:数据定义语言(DDL).数据操纵语言(DCL).事务控制语言(TCL).数据控制语言(DML). 1.1 数据定义语言(DDL) 建立.修改.删除数据库对象 ...

  7. Unity接入多个SDK的通用接口开发与资源管理(二)

    接着上篇,本篇对SDK接口进行封装.在开始之前,需要先了解下面知识. (1)unity与android之间的交互 unity通过下面方式调用android中的函数: 方法一: AndroidJavaC ...

  8. 突然虚拟机无法联网解决办法,且报错Failed to start LSB: Bring up/down

    使用sudo service network restart去启动网络时起不来 使用systemctl status network.service查看网络状态也是failed,且报错Failed t ...

  9. hadoop文件系统上的小文件合并-Hadoop Archives

    1. 什么是Hadoop archives Hadoop archives是特殊的档案格式.一个Hadoop archive对应一个文件系统目录. Hadoop archive的扩展名是.har.Ha ...

  10. Linux实战(20):非docker部署ELK

    部署环境: Elasticsearch:7.5.2 Kibana:7.5.2 Logstash:7.5.2 filebeat:7.5.2 redis:最新版 部署方式:rpm+二进制包 使用架构 软件 ...