【算法学习】Fhq-Treap(无旋Treap)

Treap——大名鼎鼎的随机二叉查找树,以优异的性能和简单的实现在OIer们中广泛流传。

这篇blog介绍一种不需要旋转操作来维护的Treap,即无旋Treap,也称Fhq-Treap。

它的巧妙之处在于只需要分离和合并两种基本操作,就能实现任意的平衡树常用修改操作。

而不需要旋转的特性也使编写代码时不需要考虑多种情况和复杂的父亲儿子关系的更新,同时降低了时间复杂度。

此外,它还可以方便地支持可持久化,实在是功能强大。

接下来系统性地介绍一下无旋Treap的原理和实现,最后讲解一下应用和例题。

一、Treap是啥?

Treap是一棵二叉查找树,满足中序遍历始终是有序的。

Treap的每个节点除了保存信息外,还要保存一个值\(pri\)。

这个\(pri\)存储的是这个节点的优先级。

它是干嘛用的呢?

实际上,Treap除了是一个二叉查找树之外,它在另一个意义下还是一个堆。

每个节点的子节点,一定要满足子节点的\(pri\)比父节点的\(pri\)小。\(pri\)越大的点越往上放。

这有什么意义呢?为什么要这样定义?

事实上,\(pri\)值是程序随机指派的,每个点的\(pri\)值是与这个点的权值无关的,是随机的。

这样随机化就可以保证Treap的深度是\(O(log\;n)\),这就是随机化的力量。

而且可以保证这样建树一定不会出现矛盾,现在模拟一下建树的过程:

先把所有节点按照中序遍历排好,然后找到其中\(pri\)最大的,把它作为整个Treap的根,左边的节点形成左子树,右边的节点形成右子树。再对左右子树递归处理。

这样最终就能建好一棵Treap。

二、Treap的基本操作

讲完了Treap的定义,来看看Treap的两个基本操作:

分离(Split)和合并(Merge)。

分离:指的是将一棵Treap按照中序遍历的顺序,分割成左右两半,满足左右两半组成的Treap的所有值都不变。

合并:指的是将两棵Treap(一般是从原先的TreapSplit出来的)合并在一起,按照中序遍历的顺序,并且所有节点的值都不变。

Split操作比较简单,先讲讲如何实现:当然,Split之前要先指定一个值k,表示Split出这个Treap的中序遍历中的前k个数作为第一棵Split出的Treap。

从这个Treap的根开始,看它的左子树的大小是否大于等于k,如果是,那么说明右子树和根都在第二棵中,继续递归到左子树中。

如果不是,那么说明左子树和根都在第一棵Treao中,继续递归到右子树中,而且k要减去左子树的大小加一。

代码:

void Split(int rt,int k,int&rt1,int&rt2){
	if(!rt) {rt1=rt2=0; return;}
	if(k<=siz[ls[rt]]){
		Split(ls[rt],k,rt1,rt2);
		ls[rt]=rt2;
		combine(rt);
		rt2=rt;
	}
	else{
		Split(rs[rt],k-siz[ls[rt]]-1,rt1,rt2);
		rs[rt]=rt1;
		combine(rt);
		rt1=rt;
	}
}

其中,rt是根,k是分出的第一棵子树的大小,rt1和rt2用来返回。

combine函数用来维护节点的大小。

接下来看Merge操作:

有两棵Treap,假设要把第二棵接到第一棵后面,那么应该怎么合并呢?

考虑两个根节点的\(pri\)值,因为第一棵在第二棵前面,所以要不然rt1(第一棵的根)在rt2(第二棵的根)的左子树,要不然rt2在rt1的右子树。

但是因为有了\(pri\)的影响,所以只能rt1和rt2中\(pri\)较大的那个作为根。

如果rt1为根,那么有rt1的右子树和rt2合并作为rt1的现在的右子树。

如果rt2为根,那么有rt2的左子树和rt1合并作为rt2的现在的左子树。

两种情况都递归进子树中即可。

代码如下:

int Merge(int rt1,int rt2){
	if(!rt1) return rt2;
	if(!rt2) return rt1;
	if(pri[rt1]<pri[rt2]){
		rs[rt1]=Merge(rs[rt1],rt2);
		combine(rt1);
		return rt1;
	}
	else{
		ls[rt2]=Merge(rt1,ls[rt2]);
		combine(rt2);
		return rt2;
	}
}

这就是两种Treap的基本操作,实现时要注意当树为空时,做特殊处理。

三、主要操作

看完了这两种操作,你肯定会问:这有个鬼用?

没法插入删除查询前驱后继排名位置的平衡树,只能分分合合的平衡树,我才不要!

但是,以上的所有操作,其实都能通过这两种基本操作实现:

查询小于等于val的数的个数:考察根节点的值和val,如果val小于根节点,递归进左子树,否则递归进右子树。

这个查询操作命名为Rank,代码:

int Rank(int rt,int v){
	if(!rt) return 0;
	if(v<val[rt]) return Rank(ls[rt],v);
	else return siz[ls[rt]]+Rank(rs[rt],v)+1;
}

插入val:先查询Rank(val),然后按照rank把整个TreapSplit成两个,把val做成一个新节点,Merge到里面即可。

void Insert(int v){
	val[++cnt]=v, pri[cnt]=ran(), siz[cnt]=1;
	int rank=Rank(Root,v);
	int rt1,rt2;
	Split(Root,rank,rt1,rt2);
	Root=Merge(Merge(rt1,cnt),rt2);
}

删除val:先查询Rank(val),然后按照rank把整个TreapSplit成三个,删除需要的点,最后Merge剩下两个。

void Delete(int v){
	int rank=Rank(Root,v);
	int rt1,rt2,rt3,tmp;
	Split(Root,rank,rt1,rt2);
	Split(rt1,rank-1,rt3,tmp);
	Root=Merge(rt3,rt2);
	// Memory Recycle?
}

查询第K个值:把整个TreapSplit成三个,输出需要的值,最后合并起来。

int Kth(int k){
	int rt1,rt2,rt3,c;
	Split(Root,k,rt1,rt2);
	Split(rt1,k-1,rt3,c);
	Root=Merge(rt3,Merge(c,rt2));
	return val[c];
}

前驱:Kth(Rank(val-1))。

后继:Kth(Rank(val)+1)。

还有很多很多操作,供大家脑补。

四、区间操作

无旋Treap和旋转Treap的更重要区别是,无旋Treap可以很方便地支持区间的操作。

如何支持?你已经看到了,Split操作分理出的就是一个个区间啊!把一整棵Treap分离出来一段区间,在上面尽情地修改吧。不过要记得像线段树写好区间pushdown哦!

比如区间翻转,就是左右子树调换,并且打上标记。区间加减乘除更不用说。

记得在Split,Merge和Rank三个函数内部加上pushdown!

五、实战应用

有很多平衡树的题目,都能用Treap解决。

推荐几道题:

Luogu P3369 测试你的Treap普通操作的熟练程度。

Luogu P3391 区间操作的应用。

Luoge P3165 区间操作和其他技巧。

原文地址:https://www.cnblogs.com/PinkRabbit/p/8681951.html

时间: 2024-11-10 12:04:01

【算法学习】Fhq-Treap(无旋Treap)的相关文章

模板 - 数据结构 - 无旋Treap / FHQ Treap

普通平衡树: #include<bits/stdc++.h> using namespace std; typedef long long ll; #define ls(p) ch[p][0] #define rs(p) ch[p][1] const int MAXN = 100000 + 5; int val[MAXN], ch[MAXN][2], rnd[MAXN], siz[MAXN], tot, root; void Init() { tot = root = 0; } void Pu

[您有新的未分配科技点]无旋treap:从好奇到入门(例题:bzoj3224 普通平衡树)

今天我们来学习一种新的数据结构:无旋treap.它和splay一样支持区间操作,和treap一样简单易懂,同时还支持可持久化. 无旋treap的节点定义和treap一样,都要同时满足树性质和堆性质,我们还是用rand()来实现平衡 而无旋treap与treap不同的地方,也是其核心,就是它不旋转用两个新的核心函数:merge函数(合并两棵子树)和split函数(分裂出某棵树的前k个节点,并且作为一棵树返回) 首先看merge函数,它是一个递归实现的过程,先看代码: 1 Treap *merge(

[BZOJ3223]文艺平衡树 无旋Treap

3223: Tyvj 1729 文艺平衡树 Time Limit: 10 Sec  Memory Limit: 128 MB Description 您需要写一种数据结构(可参考题目标题),来维护一个有序数列,其中需要提供以下操作:翻转一个区间,例如原有序序列是5 4 3 2 1,翻转区间是[2,4]的话,结果是5 2 3 4 1 Input 第一行为n,m n表示初始序列有n个数,这个序列依次是(1,2--n-1,n)  m表示翻转操作次数接下来m行每行两个数[l,r] 数据保证 1<=l<

[Bzoj3223][Tyvj1729] 文艺平衡树(splay/无旋Treap)

题目链接:https://www.lydsy.com/JudgeOnline/problem.php?id=3223 平衡树处理区间问题的入门题目,普通平衡树那道题在维护平衡树上是以每个数的值作为维护的标准,而处理区间问题时,维护平衡树的应该是每个位置的下标,所以平衡树中序遍历时应该是当前区间的样子.例如: {1 2 3 4 5}翻转区间1 3,则中序遍历应该输出{3,2,1,4,5}. 提供splay和无旋Treap的做法. splay做法: 1 #include<bits/stdc++.h>

[BZOJ1588][HNOI2002]营业额统计 无旋Treap

[HNOI2002]营业额统计 时间限制: 5 Sec  内存限制: 162 MB 题目描述 营业额统计 Tiger最近被公司升任为营业部经理,他上任后接受公司交给的第一项任务便是统计并分析公司成立以来的营业情况. Tiger拿出了公司的账本,账本上记录了公司成立以来每天的营业额.分析营业情况是一项相当复杂的工作.由于节假日,大减价或者是其他情况的时候,营业额会出现一定的波动,当然一定的波动是能够接受的,但是在某些时候营业额突变得很高或是很低,这就证明公司此时的经营状况出现了问题.经济管理学上定

[BZOJ1503][NOI2004]郁闷的出纳员 无旋Treap

1503: [NOI2004]郁闷的出纳员 Time Limit: 5 Sec  Memory Limit: 64 MB Description OIER公司是一家大型专业化软件公司,有着数以万计的员工.作为一名出纳员,我的任务之一便是统计每位员工的工资.这本来是一份不错的工作,但是令人郁闷的是,我们的老板反复无常,经常调整员工的工资.如果他心情好,就可能把每位员工的工资加上一个相同的量.反之,如果心情不好,就可能把他们的工资扣除一个相同的量.我真不知道除了调工资他还做什么其它事情.工资的频繁调

沉迷数据结构1(treap&amp;非旋treap)

Achen大佬说不要沉迷数据结构否则智商会降低的. 从省选考完后就开始学treap,首先是自己yy了一个打了两百多行,然后debug了2个月还是3个月记不清了. 最后弃疗,去找了网上别人的代码抄了一遍. noip考完后补常规的一段时间,羡慕Achen能20分钟打出一个treap模板,于是自己也开始走上打板子的不归路. 到了后来可以10分钟左右打出一个结构体版的treap,看了Achen的数组版treap,觉得自己结构体版的太不优秀啦,于是就换成数组版的. 然后现在有几周没有碰过treap,感觉又

【模板】无旋Treap(FHQ)

如题,这是一个模板... 1 #include <algorithm> 2 #include <iostream> 3 #include <cstring> 4 #include <cstdio> 5 #include <cctype> 6 7 inline void read(int & x) 8 { 9 x = 0; 10 int k = 1; 11 char c = getchar(); 12 while (!isdigit(c))

Luogu 1177 - 【模板】快速排序 - [快速排序][归并排序][无旋Treap]

题目链接:https://www.luogu.org/problemnew/show/P1177 题意:输入 $n$ 以及后续 $n$ 个整数,让你将这 $n$ 个整数从小到大排序输出. 归并排序(用时: 121ms / 内存: 1568KB): #include<bits/stdc++.h> using namespace std; const int maxn=100000+5; int n,a[maxn],t[maxn]; void Merge(int l,int m,int r) {