重点:合并
理解:将b变成a的子树,按节点大小顺序,优先放右边。
回忆堆的排序:
保证左右子节点大于(小于)根节点
其中左节点小于右节点。
所以先将要合并的两个节点按大小换位,
将b合并到a的右子树去。
重复这个思路。
返回当前循环后,按照dist值交换左右子树,确报右子树距离大于左子树
更新父节点距离
返回……
#include<bits/stdc++.h>
using namespace std;
const int N = 1e5 + 10;
bool st[N];
int p[N], son[N][2], w[N], dist[N];
int n, m;
int find(int x) {
return x == p[x] ? x : p[x] = find(p[x]);
}
int mg(int x, int y) {
if(!x) return y;
if(!y) return x;
if(w[x] > w[y]) swap(x, y);
son[x][1] = mg(son[x][1], y);
if(dist[son[x][1]] > dist[son[x][0]]) swap(son[x][0], son[x][1]);
dist[x] = dist[son[x][1]] + 1;
return x;
}
int main() {
cin >> n >> m;
for(int i = 1; i <= n; i ++)
scanf("%d", &w[i]), p[i] = i;
int op, x, y, u;
for(int i = 1; i <= m; i ++) {
cin >> op;
if(op == 1) {
cin >> x >> y;
if(st[x] || st[y]) continue;
x = find(x), y = find(y);
if(x != y)
p[y] = p[x] = mg(x, y);
}
else {
cin >> x;
if(st[x]) {
cout << "-1" << endl;
continue;
}
u = find(x);
cout << w[u] << endl;
st[u] = 1;
int a = son[u][0], b = son[u][1];
//这里p[u]也赋值为p[a]。。
p[u] = p[a] = p[b] = mg(a, b);
}
}
return 0;
}