并查集(Union-Find)

并查集是一种数据结构,主要用于解决一些元素分组的问题。它支持两种操作:查找和合并。查找操作用于判断两个元素是否属于同一组,合并操作用于将两个元素所在的组进行合并。并查集可以用于解决一些经典的算法问题,如最小生成树、网络连通性等。

一、并查集的实现原理

并查集的基本思想是:每个元素都有一个父节点,如果两个元素的父节点相同,那么它们就属于同一个集合。初始时,每个元素都是一个独立的集合,其父节点就是自己。

  1. 查找操作:查找元素所在集合的代表元素(根节点)。从给定元素开始,沿着父节点链一直向上查找,直到找到根节点。在查找过程中,可以将路径上的所有元素的父节点都直接指向根节点,以优化后续查找操作。

  2. 合并操作:将两个元素所在的集合进行合并。首先分别查找两个元素所在的根节点,然后将其中一个根节点的父节点设置为另一个根节点。

二、代码实现:

cs 复制代码
#include <stdio.h>

// 定义并查集结构体
typedef struct {
    int parent; // 父节点
    int rank;   // 树的高度
} UnionFind;

// 初始化并查集
void initUnionFind(UnionFind *uf, int n) {
    for (int i = 0; i < n; i++) {
        uf[i].parent = i;
        uf[i].rank = 0;
    }
}

// 查找操作,返回元素的根节点
int find(UnionFind *uf, int x) {
    if (uf[x].parent != x) {
        uf[x].parent = find(uf, uf[x].parent); // 路径压缩
    }
    return uf[x].parent;
}

// 合并操作,将两个元素所在的集合进行合并
void unionSet(UnionFind *uf, int x, int y) {
    int rootX = find(uf, x);
    int rootY = find(uf, y);
    if (rootX == rootY) {
        return;
    }
    if (uf[rootX].rank > uf[rootY].rank) {
        uf[rootY].parent = rootX;
    } else if (uf[rootX].rank < uf[rootY].rank) {
        uf[rootX].parent = rootY;
    } else {
        uf[rootY].parent = rootX;
        uf[rootX].rank++;
    }
}

int main() {
    int n = 5;
    UnionFind uf[n];
    initUnionFind(uf, n);

    unionSet(uf, 0, 1);
    unionSet(uf, 2, 3);
    unionSet(uf, 3, 4);

    for (int i = 0; i < n; i++) {
        printf("Element %d: parent = %d, rank = %d
", i, uf[i].parent, uf[i].rank);
    }

    return 0;
}
相关推荐
Greedy Alg1 天前
LeetCode 142. 环形链表 II
算法
睡不醒的kun1 天前
leetcode算法刷题的第三十二天
数据结构·c++·算法·leetcode·职场和发展·贪心算法·动态规划
先做个垃圾出来………1 天前
残差连接的概念与作用
人工智能·算法·机器学习·语言模型·自然语言处理
SuperCandyXu1 天前
P3205 [HNOI2010] 合唱队-普及+/提高
c++·算法·洛谷
_OP_CHEN1 天前
数据结构(C语言篇):(十二)实现顺序结构二叉树——堆
c语言·数据结构·算法·二叉树·学习笔记··顺序结构二叉树
Yingjun Mo1 天前
1. 统计推断-基于神经网络与Langevin扩散的自适应潜变量建模与优化
人工智能·神经网络·算法·机器学习·概率论
地平线开发者1 天前
征程 6 | 灰度图部署链路介绍
算法·自动驾驶
地平线开发者1 天前
手撕大模型|KVCache 原理及代码解析
算法·自动驾驶
cellurw1 天前
EDID 数据结构解析与编辑工具:校验和计算、厂商/设备名编解码、物理地址读写、颜色与时序信息提取
数据结构
共享家95271 天前
经典动态规划题解
算法·leetcode·动态规划