HDU-1512-Monkey King

ACM模版

描述

题解

典型的左偏树！

问题的核心是当两个猴子帮派中的大佬斗争之后，这两个大佬要强壮值减半并且两个帮派进行合并。那么涉及到的操作有优先队列的删除节点，优先队列的插入节点，优先队列的合并，因为普通的优先队列并不适合合并，所以这里采用左偏树比较合适。

给每个猴子建立一棵左偏树，不断进行合并，这样的话，插入节点的操作可以和优先队列的合并操作合在一起写，都看做优先队列的合并即可。

具体的看代码，很好理解。

代码

#include <iostream>
#include <cstdio>

using namespace std;

const int MAXN = 1e5 + 10;

struct node
{
int key, dis;
int lson, rson;
int pre;
} Tree[MAXN];

int n, q;

void init(int p, int k)
{
Tree[p].key = k;
Tree[p].pre = p;
Tree[p].lson = Tree[p].rson = 0;
Tree[p].dis = p == 0 ? -1 : 0;
}

int find(int x)
{
if (Tree[x].pre == x)
{
return x;
}
else
{
return find(Tree[x].pre);
}
}

int merge(int x, int y)
{
if (x == 0)
{
return y;
}
if (y == 0)
{
return x;
}

if (Tree[x].key < Tree[y].key)
{
swap(x, y);
}
Tree[x].rson = merge(Tree[x].rson, y);
int &ls = Tree[x].lson, &rs = Tree[x].rson;
Tree[ls].pre = Tree[rs].pre = x;
if (Tree[ls].dis < Tree[rs].dis)
{
swap(ls, rs);
}
if (rs == 0)
{
Tree[x].dis = 0;
}
else
{
Tree[x].dis = Tree[rs].dis + 1;
}

return x;
}

int del(int x)
{
int ls = Tree[x].lson;
int rs = Tree[x].rson;
Tree[ls].pre = ls;
Tree[rs].pre = rs;
Tree[x].dis = Tree[x].lson = Tree[x].rson = 0;
return merge(ls, rs);
}

int solve(int x, int y)
{
Tree[x].key >>= 1;
Tree[y].key >>= 1;

int _x = del(x), _y = del(y);
_x = merge(x, _x);
_y = merge(y, _y);
x = merge(_x, _y);
return Tree[x].key;
}

int main(int argc, const char * argv[])
{
while (cin >> n)
{
int x;
for (int i = 1; i <= n; i++)
{
scanf("%d", &x);
init(i, x);
}
init(0, 0);

cin >> q;
int y;
while (q--)
{
scanf("%d%d", &x, &y);
int _x = find(x), _y = find(y);
if (_x == _y)
{
puts("-1");
}
else
{
printf("%d\n", solve(_x, _y));
}
}
}

return 0;
}