左偏树（可并堆）【模板】

题目描述

链接

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

$1xy$：将第 $x$ 个数和第 $y$ 个数所在的小根堆合并（若第 $x$ 或第 $y$ 个数已经被删除或第 $x$ 和第 $y$ 个数在用一个堆内，则无视此操作）。

$2x$：输出第 $x$ 个数所在的堆最小数，并将这个最小数删除（若有多个最小数，优先删除先输入的；若第 $x$ 个数已经被删除，则输出 $-1$ 并无视删除操作）。

输入格式

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

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

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

操作 $1$：$1xy$

操作 $2$：$2x$

输出格式

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

输入

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

输出

1
2

【数据规模】

对于 $30\%$ 的数据：$n\le 10$，$m\le 10$。
对于 $70\%$ 的数据：$n\le 10^3$ ，$m\le 10^3$ 。
对于 $100\%$ 的数据：$n\le 10^5$ ，$m\le 10^5$
初始时小根堆中的所有数都在 $int$ 范围内。

【样例解释】

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

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

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

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

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

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

故输出依次为 $1$、$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]);
		}
	}
}

这题还需要查询某个结点所在的根节点，可以再多维护一个 $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;
这行代码让 $null$ 结点的距离小于任何结点，不过去掉也几乎不影响性能。

又到了博客推荐环节：

可并堆之左偏树

【模板】左偏树（可并堆）

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