7-7 并查集【模板】

最新推荐文章于 2024-09-27 11:17:47 发布

Wziyuan

最新推荐文章于 2024-09-27 11:17:47 发布

阅读量349

点赞数 10

文章标签：算法数据结构

本文链接：https://blog.csdn.net/Wziyuan/article/details/135853529

版权

给出一个并查集，请完成合并和查询操作。

输入格式:

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

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

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

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

输出格式:

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

输入样例:

输出样例:

N
Y
N
Y

数据规模:

对于30%的数据，N<=10，M<=20；

对于70%的数据，N<=100，M<=1000；

对于100%的数据，N<=10000，M<=200000。

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

解题思路：

首先定义一个 UnionFind 类，包含以下成员变量和方法：
- parent_：存储每个节点的父节点信息，用于表示节点之间的连接关系。
- size_：存储每个节点所在集合的大小，用于在合并时进行优化。
- find(int x) 方法：通过递归找到节点 x 的根节点，并进行路径压缩。
- merge(int x, int y) 方法：将节点 x 所在的集合与节点 y 所在的集合合并，同时更新集合的大小。
创建 UnionFind 对象，并初始化节点数为 n。
遍历操作序列，根据每个操作的类型执行相应的操作：
- 如果操作类型是 1，则将节点 x 和节点 y 所在的集合合并，即调用 uf.merge(x - 1, y - 1)。
- 如果操作类型是 2，则判断节点 x 和节点 y 是否在同一集合中，即调用 uf.find(x - 1) == uf.find(y - 1)。如果在同一集合中，则输出 "Y"，否则输出 "N"。
完成遍历后，得到所有查询结果。

代码如下：

#include <iostream>
#include <vector>

class UnionFind {
public:
    UnionFind(int n) : parent_(n), size_(n, 1) {
        for (int i = 0; i < n; ++i) {
            parent_[i] = i;
        }
    }

    int find(int x) {
        if (parent_[x] != x) {
            // 路径压缩：将当前节点直接连接到根节点上
            parent_[x] = find(parent_[x]);
        }
        return parent_[x];
    }

    void merge(int x, int y) {
        int rootX = find(x);
        int rootY = find(y);

        if (rootX == rootY) {
            return;
        }

        // 按秩合并：把小的集合合并到大的集合中
        if (size_[rootX] < size_[rootY]) {
            std::swap(rootX, rootY);
        }

        parent_[rootY] = rootX;
        size_[rootX] += size_[rootY];
    }

private:
    std::vector<int> parent_;
    std::vector<int> size_;
};

int main() {
    int n, m;
    std::cin >> n >> m;

    UnionFind uf(n);

    for (int i = 0; i < m; ++i) {
        int z, x, y;
        std::cin >> z >> x >> y;

        if (z == 1) {
            uf.merge(x - 1, y - 1);
        } else {
            if (uf.find(x - 1) == uf.find(y - 1)) {
                std::cout << "Y" << std::endl;
            } else {
                std::cout << "N" << std::endl;
            }
        }
    }

    return 0;
}