银河英雄传说(带权并查集)

题意就是要维护并查集,切询问两个点是不是在一棵树上,如果是,输出他们间的数减一

#include<iostream>
#include<cstdio>
using namespace std;
int re(){
	char c=getchar();int all=0,pd=1;
	for(;c>'9'||c<'0';c=getchar()) if(c=='-') pd=-1;
	while(c>='0'&&c<='9') all=all*10+c-'0',c=getchar();return all*pd;
}const int N=3e4+5;int f[N],front[N],num[N];
int find(int x){
	if(x==f[x]) return x;
	int fa=find(f[x]);front[x]+=front[f[x]];return f[x]=fa;
}int abs(int x){return x>0?x:-x;}
int main(){
	int n=re();for(int i=1;i<=30000;i++) num[i]=1,f[i]=i;
	for(int i=1;i<=n;i++){
		char c;cin>>c;
		int x=re(),y=re();
		int r1=find(x),r2=find(y);
		if(c=='M'){
			front[r1]+=num[r2];
			f[r1]=r2;num[r2]+=num[r1];
			num[r1]=0;
		}else if(r1!=r2) printf("-1
");
		else printf("%d
",abs(front[x]-front[y])-1);
	}
}
原文地址:https://www.cnblogs.com/BLUE-EYE/p/9520264.html