【洛谷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=;
 int m,fa[N],front[N],num[N];
 inline int read(){
     int x=; char c=getchar();
     while(c<''||c>'') c=getchar();
     while(''<=c&&c<='') { x=(x<<)+(x<<)+c-''; 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=;i<=n;i++){
         fa[i]=i;
         front[i]=;
         num[i]=;
     }
     scanf("%d",&m);
     char type[];
     int x,y;
     while(m--){
         scanf("%s",type);
         x=read(); y=read();
         int fx=find(x),fy=find(y);
         if(type[]=='M'){
             fa[fx]=fy;
             front[fx]+=num[fy];　　//fx前面多了num[fy]个元素
             num[fy]+=num[fx];　　　//fy的队列增长了num[fx]
             num[fx]=;　　　　　　//fx的队列不存在了
         }
         else{
             if(fx!=fy) puts("-1");
             else
              printf("%d\n",abs(front[x]-front[y])-);
         }
     }
     return ;
 }