[HDU1512]Monkey King(左偏树)

用并查集维护猴子们的关系,强壮值用左偏树维护就行了

Code

#include <cstdio>
#include <algorithm>
#include <cstring>
#define N 100010
using namespace std;

int n,fa[N],m,fr[N];
//fr[x]维护猴子x所在集合在左偏树上的编号 

int Find(int x){return x==fa[x]?x:fa[x]=Find(fa[x]);}
namespace Lt{
	int cnt,l[N],r[N],v[N],d[N];
	inline void Init(){
		cnt=0;
		memset(l,0,sizeof(l)),memset(r,0,sizeof(r));
		memset(v,0,sizeof(v)),memset(d,0,sizeof(d));
	}
	int merge(int x,int y){
		if(!x||!y) return x+y;
		if(v[x]<v[y]) swap(x,y);
		r[x]=merge(r[x],y);
		if(d[r[x]]>d[l[x]]) swap(l[x],r[x]);
		d[x]=d[r[x]]+1;
		return x;
	}
	inline void clear(int x){l[x]=r[x]=d[x]=0;}//记得一定要清零 
}

inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

int main(){
	while(~scanf("%d",&n)){
		Lt::Init();
		for(int i=1;i<=n;++i) Lt::v[i]=read(),fa[i]=fr[i]=i;
		for(m=read();m--;){
			int u=read(),v=read(),px=Find(u),py=Find(v),hhh;
			if(px==py) puts("-1");
			else{
				fa[px]=hhh=py;
				u=fr[px],v=fr[py];
				int t1=Lt::merge(Lt::l[u],Lt::r[u]),t2=Lt::merge(Lt::l[v],Lt::r[v]);
				Lt::clear(u),Lt::v[u]/=2;
				Lt::clear(v),Lt::v[v]/=2;
				t1=Lt::merge(t1,u),t2=Lt::merge(t2,v);
				int x=Lt::merge(t1,t2);
				printf("%d\n",Lt::v[x]);
				fr[hhh]=x;
			}
		}
	}
	return 0;
}

 

posted @ 2018-06-11 19:43  void_f  阅读(157)  评论(0编辑  收藏  举报