并查集

最新推荐文章于 2024-07-15 18:06:38 发布
最新推荐文章于 2024-07-15 18:06:38 发布
阅读量448 收藏
点赞数
分类专栏： 算法 文章标签： 数据结构
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_42638946/article/details/114745013
版权

并查集

1. 并查集的原理

原理

  • 并查集支持的操作：（1）将两个集合合并；（2）查询两个元素是否在同一个集合中。

在这里插入图片描述

代码模板

在这里插入图片描述

  • C++
#include <iostream>

using namespace std;

const int N = 100010;

int p[N];

int find(int x) {
    if (p[x] != x) p[x] = find(p[x]);  // 路径压缩
    return p[x];
}

int main() {
    
    int n, m;
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++) p[i] = i;  // 并查集初始化
    
    while (m--) {
        char op[2];
        int a, b;
        scanf("%s%d%d", op, &a, &b);
        if (*op == 'M') p[find(a)] = find(b);
        else {
            if (find(a) == find(b)) puts("Yes");
            else puts("No");
        }
    }
    
    return 0;
}

2. AcWing上的并查集题目

AcWing 837. 连通块中点的数量

问题描述

分析

  • 在之前并查集的基础上，新开一个数组cnt，cnt[root]表示以root为根节点的集合数量

代码

#include <iostream>

using namespace std;

const int N = 100010;

int p[N], cnt[N];

int find(int x) {
    if (p[x] != x) p[x] = find(p[x]);
    return p[x];
}

int main() {
    
    int n, m;
    cin >> n >> m;
    for (int i = 1; i <= n; i++) p[i] = i, cnt[i] = 1;  // 初始化
    
    while (m--) {
        
        string op;
        int a, b;
        cin >> op;
        
        if (op == "C") {
            cin >> a >> b;
            a = find(a), b = find(b);
            if (a != b) {
                p[a] = b;
                cnt[b] += cnt[a];
            }
        } else if (op == "Q1") {
            cin >> a >> b;
            if (find(a) == find(b)) puts("Yes");
            else puts("No");
        } else {
            cin >> a;
            cout << cnt[find(a)] << endl;
        }
    }
    
    return 0;
}

AcWing 240. 食物链

问题描述

分析

在这里插入图片描述

代码

#include <iostream>

using namespace std;

const int N = 50010;

int p[N], d[N];  // d[i]表示i到p[i]的距离，初始为0

int find(int x) {
    if (p[x] != x) {  // 如果p[x]不是根节点，执行该函数后p[x]就是根节点
        int t = find(p[x]);
        d[x] += d[p[x]];
        p[x] = t;
    }
    return p[x];
}

int main() {
    
    int n, m;
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++) p[i] = i;
    
    int res = 0;
    while (m--) {
        int t, x, y;
        scanf("%d%d%d", &t, &x, &y);
        if (x > n || y > n) res++;
        else {
            int px = find(x), py = find(y);
            if (t == 1) {  // 表示x和y是同类
                if (px == py && (d[x] - d[y]) % 3) res++;
                else if (px != py) {  // x和y不在同一个集合中
                    p[px] = py;
                    d[px] = d[y] - d[x];
                }
            } else {  // 表示x吃y
                // 如果合法的话，d[x]一定比d[y]大
                if (px == py && (d[x] - d[y] - 1) % 3) res++;
                else if (px != py) {
                    p[px] = py;
                    d[px] = d[y] + 1 - d[x];
                }
            }
        }
    }
    printf("%d\n", res);
    
    return 0;
}

AcWing 1250. 格子游戏

问题描述

分析

  • 将二维坐标转化为一维，然后判断是否存在两个点在同一个并查集中即可

代码

#include <iostream>

using namespace std;

const int N = 40010;

int n, m;
int p[N];

// 将二维坐标转化为一维，要求从左上角为(0, 0)
int get(int x, int y) {
    return x * n + y;
}

int find(int x) {
    if (p[x] != x) p[x] = find(p[x]);
    return p[x];
}

int main() {
    
    cin >> n >> m;
    
    for (int i = 0; i < n * n; i++) p[i] = i;
    
    int res = 0;
    for (int i = 1; i <= m; i++) {
        int x, y;
        char d;
        cin >> x >> y >> d;
        x--, y--;
        int a = get(x, y);
        int b;
        if (d == 'D') b = get(x + 1, y);
        else b = get(x, y + 1);
        
        int pa = find(a), pb = find(b);
        if (pa == pb) {
            res = i;
            break;
        }
        p[pa] = pb;
    }
    
    if (!res) puts("draw");
    else cout << res << endl;
    
    return 0;
}

AcWing 1252. 搭配购买

问题描述

分析

  • 题目要求，如果买一个物品，则与其相关的所有物品必须全部买下来（无向图，不是有依赖的背包问题）。我们可以找出所有的连通块，每个连通块看成一个物品，然后采用01背包模型就可以解决这个问题。
  • 01背包可以参考：背包问题（背包九讲）

代码

#include <iostream>

using namespace std;

const int N = 10010;

int n, m, vol;
int v[N], w[N];
int p[N];
int f[N];

int find(int x) {
    if (p[x] != x) p[x] = find(p[x]);
    return p[x];
}

int main() {
    
    cin >> n >> m >> vol;
    
    for (int i = 1; i <= n; i++) p[i] = i;
    for (int i = 1; i <= n; i++) cin >> v[i] >> w[i];
    
    while (m--) {
        int a, b;
        cin >> a >> b;
        int pa = find(a), pb = find(b);
        if (pa != pb) {
            v[pb] += v[pa];
            w[pb] += w[pa];
            p[pa] = pb;
        }
    }
    
    // 01背包
    for (int i = 1; i <= n; i++)
        if (p[i] == i)  // 并查集的根节点代表这个连通块合并成的物品
            for (int j = vol; j >= v[i]; j--)
                f[j] = max(f[j], f[j - v[i]] + w[i]);
    
    cout << f[vol] << endl;
    return 0;
}

AcWing 237. 程序自动分析

问题描述

分析

在这里插入图片描述

代码

#include <iostream>
#include <unordered_map>

using namespace std;

const int N = 2000010;

int n, m;
int p[N];
unordered_map<int, int> S;  // 用于不保序的离散化

struct {
    int x, y, e;  // 原始输入的i,j和约束条件类型
} query[N];

int get(int x) {
    if (S.count(x) == 0) S[x] = ++n;
    return S[x];
}

int find(int x) {
    if (p[x] != x) p[x] = find(p[x]);
    return p[x];
}

int main() {
    
    int T;
    scanf("%d", &T);
    while (T--) {
        n = 0;
        S.clear();
        scanf("%d", &m);
        for (int i = 0; i < m; i++) {
            int x, y, e;
            scanf("%d%d%d", &x, &y, &e);
            query[i] = {get(x), get(y), e};
        }
        
        for (int i = 1; i <= n; i++) p[i] = i;
        
        // 合并所有相等约束条件
        for (int i = 0; i < m; i++)
            if (query[i].e == 1) {
                int pa = find(query[i].x), pb = find(query[i].y);
                p[pa] = pb;
            }
        
        // 检查所有不等条件
        bool has_conflict = false;
        for (int i = 0; i < m; i++)
            if (query[i].e == 0) {
                int pa = find(query[i].x), pb = find(query[i].y);
                if (pa == pb) {
                    has_conflict = true;
                    break;
                }
            }
        
        if (has_conflict) puts("NO");
        else puts("YES");
    }
    
    return 0;
}

AcWing 239. 奇偶游戏

问题描述

分析

在这里插入图片描述

代码

#include <iostream>
#include <unordered_map>

using namespace std;

const int N = 20010;

int n, m;
int p[N], d[N];
unordered_map<int, int> S;

int get(int x) {
    if (S.count(x) == 0) S[x] = ++n;
    return S[x];
}

int find(int x) {
    if (p[x] != x) {
        int root = find(p[x]);
        d[x] = d[x] ^ d[p[x]];
        p[x] = root;
    }
    return p[x];
}

int main() {
    
    cin >> n >> m;
    n = 0;
    for (int i = 0; i < N; i++) p[i] = i;
    
    int res = m;
    for (int i = 1; i <= m; i++) {
        int a, b;
        string type;
        cin >> a >> b >> type;
        a = get(a - 1), b = get(b);
        
        int t = 0;
        if (type == "odd") t = 1;
        
        int pa = find(a), pb = find(b);
        if (pa == pb) {  // 说明a和b在同一个集合中
            if ((d[a] ^ d[b]) != t) {
                res = i - 1;
                break;
            }
        } else {
            p[pa] = pb;
            d[pa] = d[a] ^ d[b] ^ t;
        }
    }
    
    cout << res << endl;
    
    return 0;
}

也可以采用下面的写法（d[x]存储x到父节点的距离，类似于食物链的做法）：

#include <iostream>
#include <unordered_map>

using namespace std;

const int N = 20010;

int n, m;
int p[N], d[N];
unordered_map<int, int> S;

int get(int x) {
    if (S.count(x) == 0) S[x] = ++n;
    return S[x];
}

int find(int x) {
    if (p[x] != x) {
        int root = find(p[x]);
        d[x] += d[p[x]];  // 可能会导致溢出
        p[x] = root;
    }
    return p[x];
}

int main() {
    
    cin >> n >> m;
    n = 0;
    for (int i = 0; i < N; i++) p[i] = i;
    
    int res = m;
    for (int i = 1; i <= m; i++) {
        int a, b;
        string type;
        cin >> a >> b >> type;
        a = get(a - 1), b = get(b);
        
        int t = 0;
        if (type == "odd") t = 1;
        
        int pa = find(a), pb = find(b);
        if (pa == pb) {  // 说明a和b在同一个集合中
            if (((d[a] + d[b]) % 2 + 2) % 2 != t) {  // 必须保证余数为负，也可以处理
                res = i - 1;
                break;
            }
        } else {
            p[pa] = pb;
            d[pa] = d[a] + d[b] + t;
        }
    }
    
    cout << res << endl;
    
    return 0;
}
  • 方法2：带扩展域的并查集

分析

在这里插入图片描述

代码

#include <iostream>
#include <unordered_map>

using namespace std;

const int N = 40010, Base = N / 2;

int n, m;
int p[N], d[N];
unordered_map<int, int> S;

int get(int x) {
    if (S.count(x) == 0) S[x] = ++n;
    return S[x];
}

int find(int x) {
    if (p[x] != x) p[x] = find(p[x]);
    return p[x];
}

int main() {
    
    cin >> n >> m;
    n = 0;
    for (int i = 0; i < N; i++) p[i] = i;
    
    int res = m;
    for (int i = 1; i <= m; i++) {
        int a, b;
        string type;
        cin >> a >> b >> type;
        a = get(a - 1), b = get(b);
        
        if (type == "even") {
            if (find(a + Base) == find(b)) {
                // 此时题目输入为a,b是同类，但是发现a,b是异类，矛盾
                res = i - 1;
                break;
            }
            p[find(a)] = find(b);
            p[find(a + Base)] = find(b + Base);
        } else {
            if (find(a) == find(b)) {
                // 此时题目输入为a,b是异类，但是发现a,b是同类，矛盾
                res = i - 1;
                break;
            }
            p[find(a + Base)] = p[find(b)];
            p[find(a)] = p[find(b + Base)];
        }
    }
    
    cout << res << endl;
    
    return 0;
}

AcWing 238. 银河英雄传说

问题描述

分析

在这里插入图片描述

代码

#include <iostream>

using namespace std;

const int N = 30010;

int m;
// s存储集合中元素的个数，p[x]表示x到p[x]的距离
int p[N], s[N], d[N];

int find(int x) {
    if (p[x] != x) {
        int root = find(p[x]);
        d[x] += d[p[x]];
        p[x] = root;
    }
    return p[x];
}

int main() {
    
    scanf("%d", &m);
    
    for (int i = 0; i < N; i++) {
        p[i] = i;
        s[i] = 1;
        // d默认全为0，不需要初始化了
    }
    
    while (m--) {
        char op[2];
        int a, b;
        scanf("%s%d%d", op, &a, &b);
        if (op[0] == 'M') {
            int pa = find(a), pb = find(b);
            d[pa] = s[pb];
            s[pb] += s[pa];
            p[pa] = pb;
        } else {
            int pa = find(a), pb = find(b);
            if (pa != pb) puts("-1");
            else printf("%d\n", max(0, abs(d[a] - d[b]) - 1));
        }
    }
    
    return 0;
}

3. 力扣上的并查集题目

你好世界wxx
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
并查集 (Union-Find Sets)及其应用
fisher_jiang的专栏
08-25 1万+
并查集 (Union-Find Sets)并查集：(union-find sets)是一种简单的用途广泛的集合. 并查集是若干个不相交集合，能够实现较快的合并和判断元素所在集合的操作，应用很多。一般采取树形结构来存储并查集，并利用一个rank数组来存储集合的深度下界，在查找操作时进行路径压缩使后续的查找操作加速。这样优化实现的并查集，空间复杂度为O(N)，建立一个集合的时间复杂度为O(1)，N
Java实现并查集
08-18
Java实现并查集 Java实现并查集是指使用Java语言实现的并查集数据结构。并查集是一种树形结构，可以高效地解决元素之间的连接和分离问题。以下是Java实现并查集的详细知识点： 一、并查集的定义 并查集是一种树形...
并查集（标准并查集、带权并查集、扩展域并查集）
最新发布
xmf_learn_c的博客
07-15 1929
并查集是一种数据结构，用来管理集合之间的关系，可以用来进行判断是否属于同一集合或将不同的集合合并等操作。我们用个生活中的例子来说明一下原本小明家这个集合只有小明，小红家只有小红，那么他们结婚后（合并）就属于一个家庭（一个集合），像这样，通过并查集我们能够合并不同的集合，然后还能快速地查找他们是不是同一个集合。
Leetcode 并查集详解
03-08
### Leetcode 并查集详解 #### 一、并查集概述 并查集（Disjoint Set）是一种高效的数据结构，主要用于处理元素的合并及查询元素间的连通性问题。在许多计算机科学领域中有着广泛的应用，特别是对于图论、连通性...
C++实现并查集
12-17
本文实例为大家分享了C++实现并查集的具体代码，供大家参考，具体内容如下 #include #include #include using namespace std; class UnionFind{ private: vector<int> parent; int count; //优化，记录p和q...
【算法与数据结构】—— 并查集
热门推荐
酱懵静的博客
03-26 15万+
并查集是一种树型的数据结构，用于处理一些不相交集合的合并及查询问题（即所谓的并、查）。比如说，我们可以用并查集来判断一个森林中有几棵树、某个节点是否属于某棵树等。并查集主要由一个整型数组pre[ ]和两个函数find( )、join( )构成。 数组 pre[ ] 记录了每个点的前驱节点是谁，函数 find(x) 用于查找指定节点 x 属于哪个集合，函数 join(x,y) 用于合并两个节点 x 和 y 。
算法基础（一）：并查集
m0_50997138的博客
04-16 56
** 并查集 **在一些有N个元素的集合应用问题中，我们通常是在开始时让每个元素构成一个单元素的集合，然后按一定顺序将属于同一组的元素所在的集合合并，其间要反复查找一个元素在哪个集合中。这一类问题近几年来反复出现在信息学的国际国内赛题中。其特点是看似并不复杂，但数据量极大，若用正常的数据结构来描述的话，往往在空间上过大，计算机无法承受；即使在空间上勉强通过，运行的时间复杂度也极高，根本就不可能在比赛规定的运行时间（1～3秒）内计算出试题需要的结果，只能用并查集来描述。 并查集是一种树型的数据结构，用于*处理
并查集详解与应用
"并查集入门" 并查集是一种用于处理集合动态合并与查询问题的数据结构，常用于解决一些具有连接关系的问题，如图的连通性判断、团伙问题等。在ACM（国际大学生程序设计竞赛）中，由于其高效的操作特性，被广泛应用...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值