银河英雄传说（带权并查集）

Posted 2020-12-31 blue-eye

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

#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);
    }
}