【AcWing 238. 银河英雄传说】带权并查集

题目链接

题意：

现在有两种操作，第一种是 M x y ，表示将第 x 个战舰所在列的全部战舰整体地挪到第 y 个战舰所在列后面，第二种是 C x y ，询问第 x 个战舰和 第 y 个战舰的距离是多少。

分析:

这个题很容易就能想到并查集，在合并区间的时候需要一个size属性记录每个集合的元素的数量，其他都是带权并查集的板子，具体看代码吧：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<vector>
#include<queue>
#include<map>
#include<stack>
#include<cstdlib>
#include<climits>
#include<unordered_map>
using namespace std;
typedef long long LL;
typedef pair<int,int> PII;
typedef unsigned long long ull;
const int mod = 1e9+7;
const int N = 30010;
int pa[N],deep[N],siz[N];
int find(int x){
	if(x != pa[x]){
//		int t = find(pa[x]);
//		deep[x] += deep[pa[x]];
//		pa[x] = t;	
		int t=pa[x];
		pa[x]=find(pa[x]);
		deep[x]+=deep[t];
	}
	return pa[x];
}
int main(){
	int T;
	scanf("%d",&T);
	for(int i=0;i<N;i++) deep[i] = 0, siz[i] = 1,pa[i] = i;
	while(T--){
		char op[2];
		int x,y;
		scanf("%s%d%d",op,&x,&y);
		int a = find(x),b = find(y);
		if(op[0] == 'M'){
			if(a == b) continue;
			pa[a] = b;
			deep[a] = siz[b];
			siz[b] += siz[a];
		}
		else{
			if(a != b) cout<<-1<<endl;
			else cout<<max(0,abs(deep[x]-deep[y])-1)<<endl;
		}
	}
	return 0;
}