洛谷P1196 银河英雄传说

题目大意：

一些战舰，初始时每个都在自己编号对应的队列中，现在有两个操作 ：
M a b：把a战舰所在的队列顺序不变的接到b战舰所在队列尾部

C a b：询问a b在同一个队列之间的距离 如果a b不在同一队列输出-1

思路：

很显然的并查集套路但是自己一开始想岔了，以为只用一个dis数组维护孩子与祖先的距离就够了，但是发现合并的时候不对劲，比如Merge(a,b)我一开始是用的dis[fb]=dis[a]+1;但是突然发现其实输入给你的是一个队列中的任意一个战舰，而我一开始是把a当做a所在队列的尾部战舰去更新的，显然这样不行 ，那么就会想到在合并的时候需要知道我要接在他上面的那个队列有多长才能根据这个改变自己的dis，于是加一个num数组记录每个队列战舰的数量即可。

代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<stack>
#include<bitset>
#include<cstdlib>
#include<cmath>
#include<set>
#include<list>
#include<deque>
#include<map>
#include<queue>
using namespace std;
typedef long long ll;
const double PI = acos(-1.0);
const double eps = 1e-6;
const int INF = 1000000000;
const int maxn = 34567;
int T,n,m;
int f[maxn];
int dis[maxn];//与根的距离
int num[maxn];//该队列中的总战舰数

void init()
{
    for(int i=0;i<=30000;i++)
    {
        f[i]=i;
        dis[i]=0;
        num[i]=1;
    }
}

int Get_f(int x)//并查集路径压缩
{
    if(f[x]==x)
        return x;
    else
    {
        int fa=Get_f(f[x]);
        dis[x]+=dis[f[x]];
        return f[x]=fa;
    }
}

void Merge(int v,int u)//合并
{
    int vv=Get_f(v);
    int uu=Get_f(u);
    if(vv!=uu)
    {
        f[vv]=uu;
        dis[vv]=num[uu];
        num[uu]+=num[vv];
        num[vv]=0;
    }
}

int main()
{
    init();
    scanf("%d",&T);
    char op[2];
    int a,b;
    while(T--)
    {
        scanf("%s%d%d",op,&a,&b);

        if(op[0]=='M')
        {
            Merge(a,b);
        }

        else
        {
            int aa=Get_f(a);
            int bb=Get_f(b);
            if(aa!=bb)
                printf("-1\n");
            else
                printf("%d\n",abs(dis[a]-dis[b])-1);
        }
    }
    return 0;
}