COGS-526 爱争吵的猴子（左偏树）

526. 爱争吵的猴子

★★★   输入文件： monkeyk.in   输出文件： monkeyk.out    简单对比
时间限制：1 s   内存限制：128 MB

【问题描述】

在一个森林里,住着N只好斗的猴子.开始,他们各自为政,互不相干.但是猴子们不能消除争吵,但这仅仅发生在两只互不认识的猴子之间.当争吵发生时,争吵的两只猴子都会求助他们各自最强壮的朋友,并且决斗.当然,决斗之后,两只猴子及他们所有的朋友都相互认识了,并且成为朋友,争吵将不会在他们之间发生.

假定每一只猴子有一个强壮值,在每次决斗之后变为原来的一半(例如,10将为变为5,5将会变为2).

假定每一只猴子认识他自己. 也就是当他发生争吵,并且自己是他的朋友中最强壮的,他将代表自己进行决斗.

【输入格式】

有几组测试数据,每组测试数据由两部分构成.

第一部分:第一行有一个整数 N(N<=100,000),表示猴子的数量.下面有N行.每行有一个数,表示猴子的强壮值(<=32768).

第二部分:第一行有一个整数M(M<=100,000),表示有M次争吵发生.下面有M行,每行有两个整数x和y,表示在第x只猴子和第y只猴子之间发生争吵.

【输出格式】

对于每一次争吵,如果两只猴子认识,输出-1,否则输出一个数,表示决斗后朋友中最强壮猴子的强壮值.

【输入输出样例】

monkeyk.in

5 
20 
16 
10 
10 
4 
5 
2 3 
3 4 
3 5 
4 5 
1 5

monkeyk.out

8 
5 
5 
-1 
10 

题解：左偏树模板题，猴子发生争吵后，就将两颗树合并，每次争吵选取树的根节点。

#include<algorithm>
#include<cstdio>
#include<string.h>
using namespace std;
const int MX = 1e5 + 5;
struct node {
    int lson, rson, w, h, fa;
} t[MX];
int merge(int A, int B) {
    if (!A) return B;
    if (!B) return A;
    if (t[A].w < t[B].w) swap(A, B);
    t[A].rson = merge(t[A].rson, B);
    int ls = t[A].lson, rs = t[A].rson;
    if (t[ls].h < t[rs].h) {
        swap(ls, rs);
        swap(t[A].lson, t[A].rson);
    }
    if (t[A].rson) t[A].h = t[rs].h + 1;
    if (ls) t[ls].fa = A;
    if (rs) t[rs].fa = A;
    t[A].fa = 0;
    return A;
}
void init(int rt, int w) {
    t[rt].w = w;
    t[rt].h = -1;
    t[rt].fa = 0;
    t[rt].lson = t[rt].rson = 0;
}
int find(int x) {
    return t[x].fa == 0 ? x : (t[x].fa = find(t[x].fa));
}
int main() {
    int n, m;
    freopen("monkeyk.in","r",stdin);
    freopen("monkeyk.out","w+",stdout);
    scanf("%d", &n);
    int w;
    for (int i = 1; i <= n; i++) {
        scanf("%d", &w);
        init(i, w);
    }
    scanf("%d", &m);
    int a, b;
    while (m--) {
        scanf("%d%d", &a, &b);
        a = find(a);
        b = find(b);
        if (a == b) {
            printf("-1\n");
            continue;
        }
        t[a].w /= 2;
        t[b].w /= 2;
        int ta = merge(t[a].lson, t[a].rson);
        int tb = merge(t[b].lson, t[b].rson);
        int rt = merge(ta, tb);
        init(a, t[a].w);
        init(b, t[b].w);
        rt = merge(rt, a);
        rt = merge(rt, b);
        printf("%d\n", t[rt].w);
    }

    return 0;
}