洛谷 P3377 【模板】左偏树(左偏树模板, 模板题)

左偏树模板,并查集
本题要点:
1、题目模拟了左偏树的 合并,删除最小元素的操作。还要查询每一个节点所在的左偏树的根节点,
这样需要在结构体 添加每一个节点的父节点下标 fa。 然后在合并 操作时候,维护好 fa 的指向。
查询某节点 所在的左偏树的根节点, 类似于并查集的 get 操作,并且需要压缩路径。如果不压缩路径,
最后一个测试点TLE.
2、删除左偏树的根节点x, 需要把 tree[x].val = -INF, 然后x点的fa 指向其左右孩子合并后的根节点,
tree[x].fa = merge(tree[x].lc, tree[x].rc);
一般的左偏树删除操作,都是把 tree[x].fa 置零。 这里置零的话, 压缩路径后,get 操作会出错。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
const int MaxN = 100010, INF = 0x3f3f3f3f;
using namespace std;
int tot, n, m;

struct node
{
	int val, lc, rc, dis, fa;	//val表示值,lc合rc分别表示左子树合右子树,dis表示节点的距离,fa表示父亲
}tree[MaxN];

int merge(int x, int y)
{
	if(0 == x)	return y;
	if(0 == y)	return x;
	if(tree[x].val > tree[y].val || (tree[x].val == tree[y].val && x > y))
		swap(x, y);
	tree[x].rc = merge(tree[x].rc, y);
	tree[tree[x].rc].fa = x;
	if(tree[tree[x].rc].dis > tree[tree[x].lc].dis)
	{
		swap(tree[x].rc, tree[x].lc);
	}
	if(tree[x].rc == 0)	
	{
		tree[x].dis = 0;	
	}else{
		tree[x].dis = tree[tree[x].rc].dis + 1;
	}
	return x;
}

//删除最小节点
void del(int x)
{
	int l = tree[x].lc, r = tree[x].rc;
	tree[x].lc = tree[x].rc = tree[x].dis = 0;
	tree[x].val = -INF;
	tree[l].fa = l, tree[r].fa = r;
	tree[x].fa = merge(l, r);
}

int get(int x)
{
	if(x == tree[x].fa)
		return x;
	return tree[x].fa = get(tree[x].fa);
}

int main()
{
	scanf("%d%d", &n, &m);
	for(int i = 1; i <= n; ++i)
	{
		scanf("%d", &tree[i].val);
		tree[i].lc = tree[i].rc = tree[i].dis = 0, tree[i].fa = i;
	}
	int cmd, x, y;
	for(int i = 0; i < m; ++i)
	{
		scanf("%d", &cmd);
		if(1 == cmd)
		{
			scanf("%d%d", &x, &y);
			int fax = get(x), fay = get(y);
			if(tree[x].val == -INF || tree[y].val == -INF || fax == fay)
			{
				continue;	
			}else{
				merge(fax, fay);
			}
		}else{
			scanf("%d", &x);
			if(tree[x].val == -INF)
				printf("-1\n");
			else
			{
				int fax = get(x);
				printf("%d\n", tree[fax].val);
				del(fax);
			}
		}
	}
	return 0;
}

/*
5 5
1 5 4 2 3
1 1 5
1 2 5
2 2
1 4 2
2 2
*/

/*
1
2
*/
©️2020 CSDN 皮肤主题: 游动-白 设计师:上身试试 返回首页