寒假数据结构学习Day11

今天学习板块-----并查集

• 合并（Union）：把两个不相交的集合合并为一个集合。
• 查询（Find）：查询两个元素是否在同一个集合中。
• 并查集的重要思想在于，用集合中的一个元素代表集合。我曾看过一个有趣的比喻，把集合比喻成帮派，而代表元素则是帮主。

• 最开始，所有大侠各自为战。他们各自的帮主自然就是自己。（对于只有一个元素的集合，代表元素自然是唯一的那个元素）

  现在1号和3号比武，假设1号赢了（这里具体谁赢暂时不重要），那么3号就认1号作帮主（合并1号和3号所在的集合，1号为代表元素）。

• 现在2号想和3号比武（合并3号和2号所在的集合），但3号表示，别跟我打，让我帮主来收拾你（合并代表元素）。不妨设这次又是1号赢了，那么2号也认1号做帮主。

• 依次类推直到结束

几个简单代码如下

初始化

int fa[MAXN];
inline void init(int n)
{
    for (int i = 1; i <= n; ++i)
        fa[i] = i;
}

查询

int find(int x)
{
    if(fa[x] == x)
        return x;
    else
        return find(fa[x]);
}

合并

inline void merge(int i, int j)
{
    fa[find(i)] = find(j);
}

不过这样一直进行下去的话会发现路径会拉的很长很长，时间上来看不断回溯，所以时间复杂度很高。所以需要路径压缩。。

int find(int x)
{
    if(x == fa[x])
        return x;
    else{
        fa[x] = find(fa[x]);  //父节点设为根节点
        return fa[x];         //返回父节点
    }
}

具体题目

洛谷P3367 【模板】并查集

题目描述

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

输入格式

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

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

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

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

输出格式

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

输入输出样例

输入 #1复制

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

输出 #1复制

N
Y
N
Y

说明/提示

对于 30%30% 的数据，�≤10N≤10，�≤20M≤20。

对于 70%70% 的数据，�≤100N≤100，�≤103M≤103。

对于 100%100% 的数据，1≤�≤1041≤N≤104，1≤�≤2×1051≤M≤2×105，1≤��,��≤�1≤Xi​,Yi​≤N，��∈{1,2}Zi​∈{1,2}。

代码实现

#include <stdio.h>

int n, m;
int pre[100000];

int find1(int x) {
    while (x != pre[x]) {
        x = pre[x];
    }
    return x;
}

int main() {
    scanf("%d %d", &n, &m);
    for (int i = 1; i <= n; i++) {
        pre[i] = i;
    }
    while (m--) {
        int op, x, y;
        scanf("%d %d %d", &op, &x, &y);
        int a, b;
        a = find1(x);
        b = find1(y);
        if (op == 1) {
            pre[a] = b;
        } else if (op == 2) {
            if (a == b) {
                printf("Y\n");
            } else {
                printf("N\n");
            }
        }
    }
    return 0;
}