算法竞赛进阶指南 0X40数据结构进阶——银河英雄传说

题目链接

Acwing 238. 银河英雄传说 easy

题目描述

有一个划分为 $N$ 列的星际战场，各列依次编号为 $1,2,\dots ,N$。

有 $N$ 艘战舰，也依次编号为 $1,2,\dots ,N$，其中第 $i$ 号战舰处于第 $i$ 列。

有 T 条指令，每条指令格式为以下两种之一：

• M i j，表示让第 $i$ 号战舰所在列的全部战舰保持原有顺序，接在 $j$ 号战舰所在列的尾部。
• 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\le 30000,T\le 500000$

输入样例：

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

输出样例：

-1
1

解法：带权值的并查集

我们用 $d[x]$ 表示，节点$x$ 到它的祖宗节点的距离。

我们用 $sz[x]$ 表示，以节点 $x$ 为根节点的节点数量。

我们用 $p[x]$ 维护一个并查集。

如果此时有一个操作 M a b，即 我们要把 $a$ 号战舰所在列的所有战舰，都放在 $b$ 号战舰的那列后面。

如果 $x,y$ 分别是 $a,b$ 的祖宗节点，即 $find(a)=x,find(b)=y$。

合并之前：

合并之后：

合并之后的一些操作：

• 此时 $x$ 和 $y$ 之间的距离就是 $sz[y]$ ， 即 $d[x]=sz[y]$；
• 以 $y$ 为根节点的集合增加了 $sz[x]$ 个节点，即 $sz[y]=sz[y]+sz[x]$；
• 两个集合合并到一起了，即 $p[x]=y$；

路径压缩：

对于一个节点 $x$，首先会递归的找到它的祖宗节点 $pa$，然后不断地更新所有节点到祖宗节点的距离，并且压缩路径。

对于查询操作 C a b，我们假设 $x=find(a),y=find(b)$。

• 如果 $a$ 和 $b$ 两艘战舰不在同一列中，即 $x\ne y$，那么直接输出 $-1$。
• 如果在同一列中，那么两艘战舰中间间隔的战舰数量为 $|d[x]-d[y]|-1$ ，由于这个值是可能为负数的（当$a=b$ 时）。所以我们最好要和 $0$ 取一个最大值，即 $max(0,|d[x]-d[y]|-1)$。

时间复杂度： $O(n)$

C++代码：

#include <iostream>

using namespace std;

const int N = 3e4+10;

int p[N] , sz[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(){
    int t;
    cin>>t;
    
    //初始化
    for(int i = 0;i < N;++i){
        p[i] = i;
        sz[i] = 1;
        d[i] = 0;
    }
    
    while(t--){
        char op[2];
        int a , b;
        scanf("%s%d%d",op,&a,&b);
        
        int x = find(a) , y = find(b);
        
        if(op[0] == 'M'){
            //两艘战舰不在同一列中才能合并
            if(x == y) continue;
            
            d[x] += sz[y];
            sz[y] += sz[x];
            p[x] = y;
        }
        else{
            //两艘战舰不在同一列直接输出 -1
            if(x != y) puts("-1");
            //否则输出间隔距离
            else cout<<max(0 , abs(d[a] - d[b]) - 1)<<'\n';
        }
    }
    return 0;
}