洛谷P1196 银河英雄传说

题目：https://www.luogu.org/problemnew/show/P1196

题意：有好多好多舰队排成一排。

M i，j表示将编号i舰队所在的列排到编号j舰队所在列的后面。

C i，j表示查询编号i舰队和编号j舰队如果在一列，中间间隔了多少只舰，如果不在同一列输出-1

思路：

很自然能想到用一个数组可以表示某一个舰排在这一列的第几个

当要查询的时候只需要用并查集查一下队首相不相同，然后把他们在队伍中的编号减一下就行。

因为要知道排在第几个，合并的时候因为是放在队尾，所以合并的时候就需要知道前面这个队伍原来的大小

所以还需要一个数组来维护队伍的大小。

剩下来就交给并查集来操作。

路径压缩的时候要把父亲的排名也加到自己身上。然后所在队伍的大小也要更新。

#include<cstdio>
#include<cstdlib>
#include<map>
#include<set>
#include<cstring>
#include<algorithm>
#include<vector>
#include<cmath> 
#include<stack>
#include<queue>
#include<iostream>

#define inf 0x7fffffff
using namespace std;
typedef long long LL;
typedef pair<string, string> pr;

int t;
const int maxn = 3e4 + 5;
int par[maxn], rnk[maxn], num[maxn];

int ffind(int x)
{
    if(par[x] == x)return x;
    int tmp = par[x];
    par[x] = ffind(par[x]);
    rnk[x] += rnk[tmp];
    num[x] = num[par[x]];
    return par[x];
}

void init()
{
    for(int i = 0; i < maxn; i++){
        par[i] = i;
        rnk[i] = 0;
        num[i] = 1;
    }
}

void uunion(int x, int y)
{
    int fx = ffind(x), fy = ffind(y);
    par[fx] = fy;
    rnk[fx] += num[fy];
    num[fx] += num[fy];
    num[fy] = num[fx];
}

int main()
{
    scanf("%d", &t);
    init();
    for(int i = 0; i < t; i++){
        char type;
        int x, y;
        //getchar();
        cin>>type;
        scanf("%d %d", &x, &y);
        if(type == 'M'){
            uunion(x, y);
        }
        else{
            int fx = ffind(x), fy = ffind(y);
            if(fx != fy){
                printf("-1
");
            }
            else{
                //cout<<cnt[x]<<endl<<cnt[y]<<endl;
                printf("%d
", abs(rnk[x] - rnk[y]) - 1);
            }
        }
    }
    return 0; 
}