acwing 238 银河英雄传说（并查集）

题目地址
维护size和距离的并查集。

用d[i]表示i到父节点的距离。
在合并的时候，a的祖宗到b的祖宗的距离就是b这个集合的size。我们只需要维护两个祖宗之间的距离就行了。 而a那个集合的子节点到b的祖宗节点的距离会在find函数里面更新。

求两个点在不在同一列就是求是否在同一个集合，它们之间隔的距离就是它们分别到父节点的距离之差减1.

#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#include<bits/stdc++.h>
#define int long long
using namespace std;
typedef pair<int,int> pii;
typedef long long ll;
 
const int INF = 0x3f3f3f3f;
const double eps = 1e-5;
const int mod = 1000000007;
const int N = 1e5+10;

int fa[N],d[N],si[N];
int find(int x){
	if(x != fa[x]){
		int t = find(fa[x]);
		d[x] += d[fa[x]];
		fa[x] = t;
	}
	return fa[x];
}

void merge(int a,int b){
	a = find(a),b = find(b);
	fa[a]= b; d[a] = si[b];
	si[b] += si[a];
}

signed main(){
    IOS;
    #ifdef ddgo
        freopen("C:\\Users\\asus\\Desktop\\ddgoin.txt","r",stdin);
    #endif
    
   	int n;cin>>n;
   	for(int i=1;i<N;i++) fa[i] = i,si[i] = 1;
	for(int i=0;i<n;i++){
		string s;
		int a,b; cin>>s>>a>>b;
		if(s[0] == 'M') merge(a,b);
		else cout<<(find(a) == find(b)?abs(d[a]-d[b])-1:-1)<<endl;
	}
    
    return 0;
}