题目地址
维护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;
}