左偏树(可并堆)【模板】

题目描述

链接

如题,一开始有 n n n 个小根堆,每个堆包含且仅包含一个数。接下来需要支持两种操作:

1   x   y 1~ x~ y 1 x y:将第 x x x 个数和第 y y y 个数所在的小根堆合并(若第 x x x 或第 y y y 个数已经被删除或第 x x x 和第 y y y 个数在用一个堆内,则无视此操作)。

2   x 2~ x 2 x:输出第 x x x 个数所在的堆最小数,并将这个最小数删除(若有多个最小数,优先删除先输入的;若第 x x x 个数已经被删除,则输出 − 1 -1 −1 并无视删除操作)。

输入格式

第一行包含两个正整数 n , m n, m n,m 分别表示一开始小根堆的个数和接下来操作的个数。

第二行包含 n n n 个正整数,其中第 i i i 个正整数表示第 i i i 个小根堆初始时包含且仅包含的数。

接下来 m m m 行每行 2 2 2 个或 3 3 3 个正整数,表示一条操作,格式如下:

操作 1 1 1: 1   x   y 1~ x~ y 1 x y

操作 2 2 2: 2   x 2~ x 2 x

输出格式

输出包含若干行整数,分别依次对应每一个操作 2 2 2 所得的结果。

输入

5 5
1 5 4 2 3
1 1 5
1 2 5
2 2
1 4 2
2 2

输出

1
2

【数据规模】

对于 30 % 30\% 30% 的数据: n ≤ 10 n\le 10 n≤10, m ≤ 10 m\le 10 m≤10。
对于 70 % 70\% 70% 的数据: n ≤ 1 0 3 n\le 10^3 n≤103 , m ≤ 1 0 3 m\le 10^3 m≤103 。
对于 100 % 100\% 100% 的数据: n ≤ 1 0 5 n\le 10^5 n≤105 , m ≤ 1 0 5 m\le 10^5 m≤105
初始时小根堆中的所有数都在 i n t int int 范围内。

【样例解释】

初始状态下,五个小根堆分别为: { 1 } \{1\} {1}、 { 5 } \{5\} {5}、 { 4 } \{4\} {4}、 { 2 } \{2\} {2}、 { 3 } \{3\} {3}。

第一次操作,将第 1 1 1 个数所在的小根堆与第 5 5 5 个数所在的小根堆合并,故变为四个小根堆: { 1 , 3 } \{1,3\} {1,3}、 { 5 } \{5\} {5}、 { 4 } \{4\} {4}、 { 2 } \{2\} {2}。

第二次操作,将第 2 2 2 个数所在的小根堆与第 5 5 5 个数所在的小根堆合并,故变为三个小根堆: { 1 , 3 , 5 } \{1,3,5\} {1,3,5}、 { 4 } \{4\} {4}、 { 2 } \{2\} {2}。

第三次操作,将第 2 2 2 个数所在的小根堆的最小值输出并删除,故输出 1 1 1,第一个数被删除,三个小根堆为: { 3 , 5 } \{3,5\} {3,5}、 { 4 } \{4\} {4}、 { 2 } \{2\} {2}。

第四次操作,将第 4 4 4 个数所在的小根堆与第 2 2 2 个数所在的小根堆合并,故变为两个小根堆: { 2 , 3 , 5 } \{2,3,5\} {2,3,5}、 { 4 } \{4\} {4}。

第五次操作,将第 2 2 2 个数所在的小根堆的最小值输出并删除,故输出 2 2 2,第四个数被删除,两个小根堆为: { 3 , 5 } \{3,5\} {3,5}、 { 4 } \{4\} {4}。

故输出依次为 1 1 1、 2 2 2。

思路

需要使用一种能够动态维护最值,并且支持合并的数据结构,于是选择了左偏树。

先放代码,再说细节:

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int ls[N],rs[N],rt[N],v[N],id[N],d[N],vis[N];
int n,m,op,x,y;

int merge(int x,int y){
	if(!x||!y) return x+y;
	if(v[y]<v[x]||(v[y]==v[x]&&id[y]<id[x])) swap(x,y);
	rs[x]=merge(rs[x],y);
	if(d[ls[x]]<d[rs[x]]) swap(ls[x],rs[x]);
	d[x]=d[rs[x]]+1;
	return x;
}

int get(int x){
	if(rt[x]==x) return x;
	return rt[x]=get(rt[x]);
}

int main(){
	cin>>n>>m;
	d[0]=-1;
	for(int i=1;i<=n;i++)
		cin>>v[i],id[i]=rt[i]=i;
	while(m--){
		cin>>op>>x;
		if(op==1){
			cin>>y;
			if(vis[x]||vis[y]) continue;
			x=get(x),y=get(y);
			if(x!=y) rt[x]=rt[y]=merge(x,y);
		}else if(op==2){
			if(vis[x]){ cout<<"-1\n"; continue; }
			x=get(x);
			cout<<v[x]<<"\n";
			vis[x]=1;
			rt[ls[x]]=rt[rs[x]]=rt[x]=merge(ls[x],rs[x]);
		}
	}
}

这题还需要查询某个结点所在的根节点,可以再多维护一个 f a fa fa 关系。但是,左偏树不会保证平衡,若是一层一层往上找根节点,复杂度没有保证。
所以用并查集压缩路径的方法维护根节点。

rt[ls[x]]=rt[rs[x]]=rt[x]=merge(ls[x],rs[x]);
这行代码绝对不能写成:
rt[ls[x]]=rt[rs[x]]=merge(ls[x],rs[x]);
因为压缩路径后,x为根的子树中,很多结点的根节点直接指向x,第二种写法只修改了它左右儿子的根节点,显然是错的。
如果要一个个去修改子树结点的根,也太慢了,所以 rt[x]=merge(ls[x],rs[x]) 继续交给并查集去处理。

if(!x||!y) return x+y;
这个相当于:
if(!x) return y; if(!y) return x;

d[0]=-1;
这行代码让 n u l l null null 结点的距离小于任何结点,不过去掉也几乎不影响性能。

又到了博客推荐环节:

可并堆之左偏树

【模板】左偏树(可并堆)

数据结构: 可合并堆-左偏树

上一篇:rt-thread源码分析篇十八:rt_system_scheduler_start函数分析


下一篇:一些递归