左偏树模板,并查集
本题要点:
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
*/