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

时间:2022-01-04 13:53:31

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,如果是,那么说明右子树和根都在第二棵中,继续递归到左子树中。

如果不是,那么说明左子树和根都在第一棵Treap中,继续递归到右子树中,而且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 区间操作和其他技巧。