左偏樹學習筆記 & luogu P3377 【模板】左偏樹(可並堆)

背景:

真的不想出題,腦袋好疼。
好久之前學的了,現在複習過。

題目傳送門:

https://www.luogu.org/problem/P3377

思路:

左偏樹顧名思義就是左子樹比右子樹大的樹。
這裏的樹本質上是一個二叉堆,而且是可並堆。
具體可見:https://blog.csdn.net/a_forever_dream/article/details/85232520

代碼:

還是貼自己的吧。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
	int n,m;
	int val[100010],dis[100010],ls[100010],rs[100010],fa[100010];
int find(int x)
{
	return fa[x]==x?x:find(fa[x]);
}
int merge(int x,int y)
{
	if(!x) return y;
	if(!y) return x;
	if((val[x]>val[y])||(val[x]==val[y]&&x>y)) swap(x,y);
	rs[x]=merge(rs[x],y);
	fa[rs[x]]=x;
	if(dis[ls[x]]<dis[rs[x]]) swap(ls[x],rs[x]);
	dis[x]=dis[rs[x]]+1;
	return x;
}
int del(int x)
{
	int lc=ls[x],rc=rs[x];
	val[x]=-1;
	dis[x]=ls[x]=rs[x]=0;
	fa[lc]=lc,fa[rc]=rc;
	return merge(lc,rc);
}
int main()
{
	int t,x,y;
	scanf("%d %d",&n,&m);
	dis[0]=-1;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&val[i]);
		fa[i]=i;
	}
	for(int i=1;i<=m;i++)
	{
		scanf("%d",&t);
		if(t==1)
		{
			scanf("%d %d",&x,&y);
			if(val[x]==-1||val[y]==-1||x==y) continue;
			int t1=find(x),t2=find(y);
			if(t1==t2) continue;
			merge(t1,t2);
		}
		else
		{
			scanf("%d",&x);
			if(val[x]==-1) {printf("-1\n");continue;}
			int t1=find(x);
			printf("%d\n",val[t1]);
			del(t1);
		}
	}
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章