Monkey King
题目大意
给你一些点,一开始没有边,有 m 次操作每次选两个点,如果已经连通就输出 -1,否则连边,而且把当前连通块子树最大点除二并输出。
思路
考虑直接维护过程: 找到当前连通块最大可以用堆,那除二也就都可以维护。 那问题就是连边,就是两个连通块合并,那就是合并堆,用左偏树即可。
代码
#include<cstdio>
#include<algorithm>
using namespace std;
const int N = 1e5 + 10;
int n, a[N], m, x, y;
int fa[N], rt[N];
int find(int now) {
if (fa[now] == now) return now;
return fa[now] = find(fa[now]);
}
struct Left {
int l[N], r[N], dis[N], f[N];
void Init() {
for (int i = 1; i <= n; i++) l[i] = r[i] = dis[i] = 0, f[i] = a[i];
}
int merge(int x, int y) {
if (!x || !y) return x + y;
if (f[x] < f[y]) swap(x, y);
r[x] = merge(r[x], y);
if (dis[r[x]] > dis[l[x]]) swap(l[x], r[x]);
dis[x] = dis[r[x]] + 1;
return x;
}
void Top_half(int &x) {
f[x] /= 2; printf("%d\n", f[x]);
int L = l[x], R = r[x];
l[x] = r[x] = 0;
x = merge(L, merge(x, R));
}
}T;
int main() {
while (scanf("%d", &n) != EOF) {
for (int i = 1; i <= n; i++) scanf("%d", &a[i]), fa[i] = i, rt[i] = i;
T.Init();
scanf("%d", &m);
while (m--) {
int x, y; scanf("%d %d", &x, &y);
if (find(x) == find(y)) printf("-1\n");
else {
rt[find(x)] = T.merge(rt[find(x)], rt[find(y)]);
fa[find(y)] = find(x);
T.Top_half(rt[find(x)]);
}
}
}
return 0;
}
|