【洛谷P1196】[NOI2002]银河英雄传说

银河英雄传说

题目链接

并查集时记录下以i为首的队列的长度(如果存在这个队列)num[i]，便于合并，

和点i到队首的距离front[i]，便于查询(在find时维护)

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define N 30010
#define abs(x) ((x)>0?(x):-(x))
const int n=30000;
int m,fa[N],front[N],num[N];
inline int read(){
    int x=0; char c=getchar();
    while(c<'0'||c>'9') c=getchar();
    while('0'<=c&&c<='9') { x=(x<<3)+(x<<1)+c-'0'; c=getchar(); }
    return x;
}
inline int find(int x){
    if(fa[x]==x) return x;
    int fx=find(fa[x]);
    front[x]+=front[fa[x]];　　//改变fa[x]为祖先的同时要把fa[x]到祖先的长度加上
    return fa[x]=fx;
}
int main()
{
    for(int i=1;i<=n;i++){
        fa[i]=i;
        front[i]=0;
        num[i]=1;
    }
    scanf("%d",&m);
    char type[2];
    int x,y;
    while(m--){
        scanf("%s",type);
        x=read(); y=read();
        int fx=find(x),fy=find(y);
        if(type[0]=='M'){
            fa[fx]=fy;
            front[fx]+=num[fy];　　//fx前面多了num[fy]个元素
            num[fy]+=num[fx];　　　//fy的队列增长了num[fx]
            num[fx]=0;　　　　　　//fx的队列不存在了
        }
        else{
            if(fx!=fy) puts("-1");
            else
             printf("%d\n",abs(front[x]-front[y])-1);
        }
    }
    return 0;
}