银河英雄传说

题目链接：https://www.luogu.com.cn/problem/P1196

题意：

两种操作：M i j 表示将i所在战舰组移动到j所在战舰组后面

C i j 查询i，j之间的战舰数量

思路：

带权并查集，分别开两个数组siz[i],dist[i]记录战舰组战舰数量和战舰到战舰头的距离

find时，除了更新fa[x]还要更新 dist[find(x)]=siz[y],dist[x]+=dist[fa[x]],siz[y]+=siz[x]

merge时，更新其根节点的dist和fa以及siz

#include<bits/stdc++.h>
#define rep(i,a,n) for(int i=a;i<=n;i++)
#define pb push_back
#define endl "\n"
#define fi first
#define int long long
#define se second
//#pragma GCC optimize(3)
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef __int128 lll;
typedef pair<int,int> pii;
const int inf=0x3f3f3f3f;
const ll llmax=LLONG_MAX;
const int maxn=3e4+5;
const int mod=1e9+7;
int fa[maxn];
int dist[maxn];
int siz[maxn];

int find(int x){
	if(fa[x]!=x){
		int root=find(fa[x]);
		dist[x]+=dist[fa[x]];
		fa[x]=root;
	}
	return fa[x];
}

void merge(int x,int y){
	int a=find(x),b=find(y);
	if(a==b)return;
	dist[a]=siz[b];
	siz[b]+=siz[a];
	fa[a]=b;
}

void solve(){
	int t;cin>>t;
	for(int i=1;i<=maxn;i++){
		fa[i]=i;
		siz[i]=1;
	}
	while(t--){
		char opt;cin>>opt;
		int i,j;
		if(opt=='M'){
			cin>>i>>j;
			merge(i,j);
		}else{
			cin>>i>>j;
			int a=find(i),b=find(j);
			if(a!=b){
				cout<<-1<<endl;
			}else{
				cout<<abs(dist[i]-dist[j])-1<<endl;
			}
		}
	}

}

signed main()
{
	ios::sync_with_stdio(false),cin.tie(0);
	int T=1;
	
	while(T--){
	solve();
	}
	
	return 0;
}