罗马游戏-左偏树

罗马游戏-左偏树

题目描述

题解

可并堆，左偏树
左偏树推荐博客：https://blog.csdn.net/wang3312362136/article/details/80615874

代码实现

#include<bits/stdc++.h>//可并堆，左偏树 
#define M 1000009
using namespace std;
int n,m,f[M],a[M],root[M],dis[M],ch[M][2];
bool die[M];
char s[10];
int getfa(int x){
	if(f[x]==x) return x;
	else return f[x]=getfa(f[x]);
}
int merge(int x,int y){
	if(!x||!y) return x+y;
	if(a[x]>a[y]||(a[x]==a[y]&&x<y)) swap(x,y);//保证堆内的最小值为编号较小的 
	ch[x][1]=merge(ch[x][1],y);
	if(dis[ch[x][0]]<dis[ch[x][1]])
		swap(ch[x][0],ch[x][1]);
	dis[x]=dis[ch[x][1]]+1;
	return x;
}
int main(){
	int x,y;
	scanf("%d",&n);
	for(int i=1;i<=n;i++) f[i]=root[i]=i;
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	scanf("%d",&m);
	for(int i=1;i<=m;i++){
		scanf("%s",s);
		if(s[0]=='M'){
			scanf("%d%d",&x,&y);
			if(die[x]||die[y]) continue;
			int fx=getfa(x),fy=getfa(y);
			if(fx==fy) continue;
			f[fx]=fy;
			root[fy]=merge(root[fx],root[fy]);
		}
		else{
			scanf("%d",&x);
			if(die[x]) printf("0\n");
			else{
				int fx=getfa(x);
				printf("%d\n",a[root[fx]]);
				die[root[fx]]=1;
				root[fx]=merge(ch[root[fx]][0],ch[root[fx]][1]);
			}
		}
	}return 0;
} 

附上随机堆做法

#include<bits/stdc++.h>
#define M 1000009
using namespace std;
int n,m,f[M],a[M],root[M],dis[M],ch[M][2];
bool die[M];
char s[10];
int getfa(int x){
	if(f[x]==x) return x;
	else return f[x]=getfa(f[x]);
}
int merge(int x,int y){
	if(!x||!y) return x+y;
	if(a[x]>a[y]) swap(x,y);
	int opt=rand()&1;
	ch[x][opt]=merge(ch[x][opt],y);
	return x;
}
int main(){
	srand(time(0)); 
	int x,y;
	scanf("%d",&n);
	for(int i=1;i<=n;i++) f[i]=root[i]=i;
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]);
	scanf("%d",&m);
	for(int i=1;i<=m;i++){
		scanf("%s",s);
		if(s[0]=='M'){
			scanf("%d%d",&x,&y);
			if(die[x]||die[y]) continue;
			int fx=getfa(x),fy=getfa(y);
			if(fx==fy) continue;
			f[fx]=fy;
			root[fy]=merge(root[fx],root[fy]);
		}
		else{
			scanf("%d",&x);
			if(die[x]) printf("0\n");
			else{
				int fx=getfa(x);
				printf("%d\n",a[root[fx]]);
				die[root[fx]]=1;
				root[fx]=merge(ch[root[fx]][0],ch[root[fx]][1]);
			}
		}
	}return 0;
}