ZOJ 2334 Monkey King(左偏树)

题目链接:点击打开链接

思路:左偏树模板题, 需要维护集合关系, 这个可以用并查集很方便的维护, 另外需要用一个数组来维护每个点所在的左偏树编号。

细节参见代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<list>
#include<deque>
#include<map>
#include<queue>
#define Max(a,b) ((a)>(b)?(a):(b))
#define Min(a,b) ((a)<(b)?(a):(b))
using namespace std;
typedef long long ll;
typedef long double ld;
const ld eps = 1e-9, PI = 3.1415926535897932384626433832795;
const int mod = 1000000000 + 7;
const int INF = 0x3f3f3f3f;
// & 0x7FFFFFFF
const int seed = 131;
const ll INF64 = ll(1e18);
const int maxn = 3e5 + 10;
int T,n,m;
int tot, u, a, b, v[maxn], l[maxn], r[maxn], d[maxn], p[maxn], id[maxn];
int _find(int x) { return p[x] == x ? x : p[x] = _find(p[x]); }
int Merge(int x, int y) {
    if(!x) return y;
    if(!y) return x;
    if(v[x] < v[y]) swap(x, y);
    r[x] = Merge(r[x], y);
    if(d[l[x]] < d[r[x]]) swap(l[x], r[x]);
    d[x] = d[r[x]] + 1;
    return x;
}
int init(int x) {
    tot++;
    v[tot] = x;
    l[tot] = r[tot] = d[tot] = 0;
    return tot;
}
int Insert(int x, int y) {
    return Merge(x, init(y));
}
int top(int x) {
    return v[x];
}
int pop(int x) {
    return Merge(l[x], r[x]);
}
int main() {
    while(~scanf("%d",&n)) {
        tot = 0;
        for(int i = 1; i <= n; i++) {
            p[i] = i;
            id[i] = i;
        }
        for(int i = 1; i <= n; i++) {
            scanf("%d",&u);
            init(u);
        }
        scanf("%d",&m);
        while(m--) {
            scanf("%d%d",&a,&b);
            int x = _find(a), y = _find(b);
            if(x == y) printf("-1\n");
            else {
                int u = top(id[x]);
                id[x] = pop(id[x]);
                id[x] = Insert(id[x], u/2);
                u = top(id[y]);
                id[y] = pop(id[y]);
                id[y] = Insert(id[y], u/2);
                p[x] = y;
                id[y] = Merge(id[x], id[y]);
                u = top(id[y]);
                printf("%d\n",u);
            }
        }
    }
    return 0;
}


你可能感兴趣的:(ZOJ,左偏树,ACM-ICPC)