AcWing 238. 银河英雄传说 带权并查集

AcWing 238. 银河英雄传说

题意

有 $n$ 个战舰和 $T$ 指令，指令有两种格式：

• M i j ，表示第 $i$ 号战舰所在列保持原序，接在第 $j$ 号战舰所在列的后面。
• C i j ，表示询问第 $i$ 号战舰和第 $j$ 号战舰之间有多少战舰，如果不在同一列则输出 $-1$ 。

（$N\le 3\cdot 10^4$，$T\le 5\cdot 10^5$）

解法

• 带权并查集模板题。

• 在并查集维护元素是否在同一个集合的基础上，维护同一集合之间元素的关系。

• $d[x]$ 表示 $x$ 和 $x$ 的父亲结点之间的关系，在本题表示两点之间的距离。

• 维护一个 $sz[x]$ 表示 $x$ 是战舰的头部，这个战舰的数量。用于合并时维护 $d[x]$ 。

代码

#include <bits/stdc++.h>
#define rep(i, a, n) for (int i = a; i <= n; ++i)
#define per(i, a, n) for (int i = n; i >= a; --i)
#ifdef LOCAL
#include "Print.h"
#define de(...) W('[', #__VA_ARGS__,"] =", __VA_ARGS__)
#else
#define de(...)
#endif
using namespace std;
typedef long long ll;
const int maxn = 3e4 + 5;
int fa[maxn], sz[maxn], d[maxn];
int findR(int x) {
    if (x == fa[x]) return x;
    int rt = findR(fa[x]);
    d[x] += d[fa[x]];
    return fa[x] = rt;
}
void link(int x, int y, int f) {
    int xx = findR(x), yy = findR(y);
    fa[xx] = yy, d[xx] += sz[yy];
    sz[yy] += sz[xx];
}
int query(int x, int y) {
    if (x == y) return 0;
    int xx = findR(x), yy = findR(y);
    if (xx != yy) return -1;
    return abs(d[x] - d[y]) - 1;
}
int case_Test() {
    int T;
    scanf("%d", &T);
    rep(i, 1, maxn - 1) fa[i] = i, sz[i] = 1;
    while (T--) {
        char op[5]; int x, y;
        scanf("%s%d%d", op + 1, &x, &y);
        if (op[1] == 'M') link(x, y, 1);
        else printf("%d\n", query(x, y));
    }
    return 0;
}
int main() {
#ifdef LOCAL
    freopen("/Users/chenjinglong/Desktop/cpp_code/in.in", "r", stdin);
    freopen("/Users/chenjinglong/Desktop/cpp_code/out.out", "w", stdout);
    clock_t start = clock();
#endif
    int _ = 1;
    // scanf("%d", &_);
    while (_--) case_Test();
#ifdef LOCAL
    printf("Time used: %.3lfs\n", (double)(clock() - start) / CLOCKS_PER_SEC);
#endif
    return 0;
}