题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1512
题意:
N个猴子开始互相不认识,而且每个猴子有一个初始的战斗力。
每次两个猴子遇到后,如果互不认识,就是找自己认识的战斗力最强的猴子打架。每个猴子打架之后战斗力就会减半。之后,他们两队人就算是认识了。
给出M个询问,不认识就输出合为一对之后的最大战斗力。认识输出-1
解法:用一个左偏树维护每一个部落,每次两个部落打架就把堆顶元素弹出来,再将元素大小减半之后再加入堆中,之后将两个堆合并就是一次打架了。
代码:
#include <bits/stdc++.h>
#define sf scanf
#define pf printf
using namespace std;
const int maxn = 100000 + 50,INF = 1000000000;
int key[maxn],dis[maxn],ch[maxn][2];
int tot_node,sroot,root;
int fa[maxn];
int Merge(int a,int b){
if(!a) return b;if(!b) return a;
if(key[a] < key[b]) swap(a,b);
ch[a][1] = Merge(ch[a][1],b);
if(dis[ch[a][0]] < dis[ch[a][1]]) swap( ch[a][0],ch[a][1] );
if(ch[a][1] == 0) dis[a] = 0;
else dis[a] = dis[ch[a][1]] + 1;
return a;
}
int Find_Fa(int x){
return fa[x] == x ? x : fa[x] = Find_Fa(fa[x]);
}
int solve(int x,int y){
int xx,yy;
int lx = ch[x][0],rx = ch[x][1],ly = ch[y][0],ry = ch[y][1];
xx = Merge(ch[x][0],ch[x][1]);
ch[x][0] = ch[x][1] = 0;key[x] = key[x] / 2;
xx = Merge(xx,x);
fa[ch[y][0]] = fa[ch[y][0]] = -1;
yy = Merge(ch[y][0],ch[y][1]);
ch[y][0] = ch[y][1] = 0;key[y] = key[y] / 2;
yy = Merge(yy,y);fa[ch[y][0]] = fa[ch[y][0]] = fa[y] = yy;
xx = Merge(xx,yy);
fa[lx] = fa[rx] = fa[y] = fa[ly] = fa[ry] = fa[x] = xx;
return key[xx];
}
int main(){
int n,m;
while( sf("%d",&n) != EOF ){
for(int i = 1;i <= n;++i){
sf("%d",&key[i]);
ch[i][0] = ch[i][1] = 0;
fa[i] = i;
dis[i] = 0;
}
sf("%d",&m);
while(m--){
int x,y;sf("%d %d",&x,&y);
x = Find_Fa(x),y = Find_Fa(y);
if(x == y) pf("-1\n");
else{
pf("%d\n",solve(x,y));
}
}
}
}