【NOI2002】银河英雄传说

#include<cstdio>
#include<cstdlib>
int T,x,y,p[30001],a[30001],r[30001]; //r[i]表示i号战舰和它的队头战舰之间的距离 
char ch[2];                           //a[i]是代表为i的队一共有多少战舰
void init(){                          //实际上，如果用r[x]表示数目的话对于只有一个结点和两个结点的情况 
    for(int i=1;i<=30000;i++)         //就不好区分，设为距离可以便于统一处理
        p[i]=i,a[i]=1;                //为什么需要a[]，因为每次UNION操作是把队列置于另一个队列的队尾 
}                                     //为了得到两个队头的距离，就需要有这样的记录 
int find(int x){                      //始终注意到，UNION本质是对“代表元素”的操作 
    if(x!=p[x]){
        int px=find(p[x]);//找到当前结点x的根节点，同时在x以上的结点都完成了r[x]的更新，包括r[p[x]] 
        r[x]+=r[p[x]];    //加上距离得到当前结点x对根节点px的距离 
        p[x]=px;
    }
    return p[x];
}
void Union(int x,int y,int px,int py){
    p[px]=py;             //直接设px指向py 
    r[px]=a[py];          //那么此时px和py之间的距离就是py队列的元素数目 
    a[py]+=a[px];         //因为实际上是把px置于py队尾，那么py队列现在的数目就是两者之和 
}
int main()
{
    init();
    scanf("%d",&T);
    while(T--){
        scanf("%s%d%d",ch,&x,&y);
        int px=find(x),py=find(y); 
        if(ch[0]=='M') Union(x,y,px,py);
        else{
            if(px!=py) printf("-1\n");
            else printf("%d\n",abs(r[x]-r[y])-1); //数目=距离-1 
        }
    }
}