算法刷题【洛谷P3367】并查集 - 模板

异想之旅：本人原创博客完全手敲，绝对非搬运，全网不可能有重复；本人无团队，仅为技术爱好者进行分享，所有内容不牵扯广告。本人所有文章仅在CSDN、掘金和个人博客（一定是异想之旅域名）发布，除此之外全部是盗文！

---

题目描述

如题，现在有一个并查集，你需要完成合并和查询操作。

输入格式

第一行包含两个整数 N,M ,表示共有 N 个元素和 M 个操作。

接下来 M 行，每行包含三个整数 Zi,Xi,Yi 。

当 Zi=1 时，将 Xi 与 Yi 所在的集合合并。

当 Zi=2 时，输出 Xi 与 Yi 是否在同一集合内，是的输出 Y ；否则输出 N 。

输出格式

对于每一个 Zi=2 的操作，都有一行输出，每行包含一个大写字母，为 Y 或者 N 。

输入输出样例

In 1

4 7
2 1 2
1 1 2
2 1 2
1 3 4
2 1 4
1 2 3
2 1 4

Out 1

N
Y
N
Y

---

题目说了是并查集模板，我也就写上模板呗，AC了~

并查集的典型例题如下：

很明显，给每个人建立一个集合来存储他的亲戚时间和空间复杂度都极高，于是我们利用一个多叉树和森林的数据结构来优化存储。

具体原理和实现讲解资源已经很多了，不做重复劳动。我推荐看这两篇文章：

• 并查集_长安某～的博客-CSDN博客
• 并查集【通俗易懂】【知识点】_不甘-CSDN博客

下面是我对本题给出的代码：

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

int f[10001];

// 递归查找
int find(int a) {
    if (f[a] != a) {
        // 路径压缩：一棵深度大的树变成宽度大的树，减少递归
        return f[a] = find(f[a]);
    } else {
        return a;
    }
}

int hb(int a, int b) {
    // 合并两集合
    f[find(b)] = find(a);
}

int main() {
    int n, m;
    cin >> n >> m;
    for (int i = 1; i <= n; i++) f[i] = i;

    int z, x, y;
    while (m--) {
        cin >> z >> x >> y;
        if (z == 1) {
            hb(x, y);
        } else {
            if (find(x) == find(y))
                cout << "Y\n";
            else
                cout << "N\n";
        }
    }

    return 0;
}

亲自试了试，虽是模板题数据不是很水，不加路径压缩过不了。

find() 函数也可以不使用递归，直接循环来写（如上面推荐的第二篇文章就是循环实现的），理解起来可能会更困难，讲解写在注释里了，大家自己看：

// 循环解法
int find(int a) {
    // a是要查询的节点，ans是返回的结果
    int ans = a;
    while (f[ans] != ans) {
        // 循环寻找a的父节点，直到找到树根
        ans = f[ans];
    }
    int i;
    while (f[a] != a) {
        // 将途径节点的父节点全部刷新为树根
        i = a;  // 暂存当前节点，以保证更改a后仍能找回当前选择的节点
        a = f[a];  // 下一个要操作的节点编号
        f[a] = ans;  // 当前节点的父节点直接更新为树根
    }
    return ans;
}