银河英雄传说-并查集

有一个划分为N列的星际战场，各列依次编号为1,2,…,N。
有N艘战舰，也依次编号为1,2,…,N,其中第i号战舰处于第i列。
有T条指令，每条指令格式为以下两种之一：
1、M i j，表示让第i号战舰所在列的全部战舰保持原有顺序，接在第j号战舰所在列的尾部。
2、C i j，表示询问第i号战舰与第j号战舰当前是否处于同一列中，如果在同一列中，它们之间间隔了多少艘战舰。
现在需要你编写一个程序，处理一系列的指令。
输入格式
第一行包含整数T，表示共有T条指令。
接下来T行，每行一个指令，指令有两种形式：M i j或C i j。
其中M和C为大写字母表示指令类型，i和j为整数，表示指令涉及的战舰编号。
输出格式
你的程序应当依次对输入的每一条指令进行分析和处理：
如果是M i j形式，则表示舰队排列发生了变化，你的程序要注意到这一点，但是不要输出任何信息；
如果是C i j形式，你的程序要输出一行，仅包含一个整数，表示在同一列上，第i号战舰与第j号战舰之间布置的战舰数目，如果第i号战舰与第j号战舰当前不在同一列上，则输出-1。
数据范围
N≤30000,T≤500000N≤30000,T≤500000
输入样例：
4
M 2 3
C 1 2
M 2 4
C 4 2

输出样例：
-1
1
思路：这道题很明显是一道并查集的题，由于最后要输出的是俩搜战舰的距离，所以这里应该维护的是每一艘战舰到祖宗节点的距离，当把一艘战舰所在的一列都移动到另一艘战舰所在队列的尾部的时候，另一艘战舰的祖宗节点所在的规模要加上转移来的规模，转移来的战舰的祖宗节点到另一艘战舰祖宗节点的距离为另一队战舰的规模，查找每一艘战舰的祖宗节点时，可以更新这一艘战舰到另外一对战舰祖宗节点的距离。
ac代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=30010;
int m;
int p[N],sizer[N],d[N];
int find(int x){
 if(x!=p[x]){
  int root=find(p[x]);
  d[x]+=d[p[x]];
  p[x]=root;
 }
 return p[x];
}
int main(){
 scanf("%d",&m);
 for(int i=0;i<=N;i++){
  sizer[i]=1;
  p[i]=i;
 }
 while(m--){
  char op[2];
  int a,b;
  scanf("%s%d%d",op,&a,&b);
  if(op[0]=='M'){
   int pa=find(a),pb=find(b);
   d[pa]=sizer[pb];
   sizer[pb]+=sizer[pa];
   p[pa]=pb;
  }
  else{
   int pa=find(a),pb=find(b);
   if(pa!=pb)   puts("-1");
   else      printf("%d\n",max(0,abs(d[a]-d[b])-1));
  }
 }
 return 0;
}