[luoguP1196] 银河英雄传说（并查集）

传送门

记录 up[x] 表示 x 上方有多少个

　　 all[x] 表示当前连通的有多少个

find 的时候 和 合并的时候 更新一下即可

——代码

 #include <cstdio>
 #include <iostream>
 #define N 30001
 #define abs(x) ((x) < 0 ? -(x) : (x))

 int T;
 int f[N], up[N], all[N];

 inline int read()
 {
     int x = , f = ;
     char ch = getchar();
     for(; !isdigit(ch); ch = getchar()) if(ch == '-') f = -;
     for(; isdigit(ch); ch = getchar()) x = (x << ) + (x << ) + ch - '';
     return x * f;
 }

 inline int find(int x)
 {
     if(x ^ f[x])
     {
         int fx = f[x];
         f[x] = find(f[x]);
         up[x] += up[fx];
     }
     return f[x];
 }

 int main()
 {
     int i, x, y, fx, fy;
     char s[];
     T = read();
     for(i = ; i < N; i++) f[i] = i, all[i] = ;
     while(T--)
     {
         scanf("%s", s);
         x = read();
         y = read();
         fx = find(x);
         fy = find(y);
         if(s[] == 'M')
         {
             f[fx] = fy;
             up[fx] += all[fy];
             all[fy] += all[fx];
         }
         else fx ^ fy ? puts("-1") : printf("%d\n", abs(up[x] - up[y]) - );
     }
     return ;
 }