银河英雄传说--AcWing238&洛谷P1196（带权并查集）

题目链接https://www.acwing.com/problem/content/description/240/

有一个划分为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≤500000

输入样例：

4
M 2 3
C 1 2
M 2 4
C 4 2

输出样例：

-1

1

---

这一题一看就应该知道是个并查集。。。但它不仅问是否在同一列，还问了i,j之间间隔了多少战舰，你们我们很容易想到在边上赋一些权值，在并查集走的时候我们只需要将它走过的边上的权值加起来就好了。

对于每一列，我们都要记录一下它们的大小，先初始化为1。在查找的时候我们可以这么写：

int find(int x)
{
    if (x!=father[x]){
        int t=father[x];
        father[x]=find(father[x]);
        val[x]+=val[t];
    }
    return father[x];
}

val[t]就相当于边上的权值。合并的时候我们将j的大小加上i所在的舰队的大小：

if (s[0]=='M'){
    if (fa==fb) continue;
    father[fa]=fb;
    val[fa]=siz[fb];
    siz[fb]+=siz[fa];
}

其他的都好理解，至于val[fa]=siz[fb]的话，由于我们将father[fa]=fb了，那么下一次查找的时候我们直接就通过father[fa]找到根节点了，没有经过之前fb的舰队，所以我们要先将siz[fb]的值转移过来，之后他会自己往上爬到之前的根节点，同时会自动加上爬过路径的权值。

以下是AC代码：

#include <bits/stdc++.h>
using namespace std;

const int mac=5e5+10;

struct node
{
    char s[2];
    int x,y;
}a[mac];

int father[30005],val[30005],siz[30005];

int find(int x)
{
    if (x!=father[x]){
        int t=father[x];
        father[x]=find(father[x]);
        val[x]+=val[t];
    }
    return father[x];
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int t;
    for (int i=1; i<=30000; i++) father[i]=i,siz[i]=1;
    cin>>t;
    int i=1;
    while (t--){
        char s[2];
        int x,y;
        cin>>s>>x>>y;
        int fa=find(x),fb=find(y);
        if (s[0]=='M'){
            if (fa==fb) continue;
            father[fa]=fb;
            val[fa]=siz[fb];
            siz[fb]+=siz[fa];
        }
        else {
            if (fa==fb) cout<<abs(val[x]-val[y])-1<<endl;
            else cout<<-1<<endl;
        }
    }
    return 0;
}