想必大家都知道一种叫做二叉搜索树这东西吧,那么我们知道,在某些特殊情况下,二叉搜索树会退化成一条链,而且如果出题人成心想卡你的话也很简单,分分钟把你(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. add forward

    adb forward tcp:34999 localabstract:Unity-com.kingsgroup.rob adb forward --list adb forward --remove ...

  2. DLX AlgorithmX

    AlgorithmX精确覆盖: https://en.wikipedia.org/wiki/Knuth's_Algorithm_X DLX的基础算法 https://zh.wikipedia.org/ ...

  3. Palindrome subsequence(区间dp+容斥)

    In mathematics, a subsequence is a sequence that can be derived from another sequence by deleting so ...

  4. 01vue.config.js

      const path = require('path'); module.exports = { // 基本路径 publicPath: process.env.NODE_ENV === 'pro ...

  5. python爬取拉勾网职位信息-python相关职位

    import requestsimport mathimport pandas as pdimport timefrom lxml import etree url = 'https://www.la ...

  6. 非构造函数方式创建DbContext实例的方法

    using Microsoft.EntityFrameworkCore;using Microsoft.EntityFrameworkCore.Design;using Microsoft.Entit ...

  7. 使用grep命令查找文件中符合”.stg.“行

    某目录下有个test.txt,内容如下: www.stg.comwwstgcom 如果我这样去查找: $ grep '.stg.' test.txtwww.stg.comwwstgcom 发现第二个匹 ...

  8. leetcode刷题-82.删除排序链表中的重复元素 II

    题目 给定一个排序链表,删除所有含有重复数字的节点,只保留原始链表中 没有重复出现 的数字. 示例 1: 输入: 1->2->3->3->4->4->5输出: 1- ...

  9. python基础:面向对象

    一.定义 面向对象是一种编程方式,此编程方式的实现是基于对 类 和 对象 的使用 类:一个种类,一个模型. 对象:指具体的东西,模型造出来的东西叫做对象. 实例:实例和对象是一样的. 实例化:实例化就 ...

  10. 如何入门Pytorch之四:搭建神经网络训练MNIST

    上一节我们学习了Pytorch优化网络的基本方法,本节我们将以MNIST数据集为例,通过搭建一个完整的神经网络,来加深对Pytorch的理解. 一.数据集 MNIST是一个非常经典的数据集,下载链接: ...