算法之并查集 C语言实现3

最新推荐文章于 2023-06-22 07:00:00 发布

Chris_秦

最新推荐文章于 2023-06-22 07:00:00 发布

阅读量3.3k

点赞数

分类专栏：算法文章标签：算法

本文链接：https://blog.csdn.net/qq_29731079/article/details/62051220

版权

算法专栏收录该内容

7 篇文章 0 订阅

订阅专栏

本文详细介绍了C语言实现并查集的数据结构——UnionFind3，包括头文件定义和C代码实现，重点讲解了并查集的核心功能和使用方法。

摘要由CSDN通过智能技术生成

头文件 UnionFind3.h

#ifndef UNIONFIND3_H_INCLUDED
#define UNIONFIND3_H_INCLUDED
#include "stdlib.h"
#include "ASSERT.h"
typedef struct{
    int* parent;
    int* rank;//rank[i]表示以i为根的集合中树的层数
    int count;
}Union;

void unionFind3Init(Union *un,int n);
void freeUnion3(Union *un);
int unionFind3(Union *un,int p);
int isConnected3(Union *un,int p,int q);
void unionElements3(Union *un,int p,int q);
#endif // UNIONFIND_H_INCLUDED

c代码 UnionFind3.c

#include "UnionFind3.h"
/*********************************************
*函数名：并查集的初始化
*********************************************/
void unionFind3Init(Union *un,int n){
    int i;
    un->count = n;
    un->parent = (int*)malloc(n*sizeof(int));
    un->rank = (int*)malloc(n*sizeof(int));
    for(i=0;i<n;i++){
        un->parent[i] = i;
        un->rank[i] = 1;
    }
}
/*********************************************
*函数名：并查集将空间释放
*********************************************/
void freeUnion3(Union *un){
    free(un->parent);
    free(un->rank);
}

/*********************************************
*函数名：返回并查集中p的ID(采用路径压缩)
*********************************************/
int unionFind3(Union *un,int p){
    assert(p >= 0 && p< un->count);
//    while(p != un->parent[p]){
//        un->parent[p] = un->parent[un->parent[p]];  //路径压缩
//        p = un->parent[p];
//    }
//    return p;

    if(p != un->parent[p]){                           //使用递归的方法进行路径压缩
        un->parent[p] = unionFind3(un,un->parent[p]);
    }
    return un->parent[p];
}
/*********************************************
*函数名：判断并查集中p和q是否是一个集合
*********************************************/
int isConnected3(Union *un,int p,int q){
    return unionFind3(un,p) == unionFind3(un,q) ;
}
/*********************************************
*函数名：并查集的合并
*********************************************/
void unionElements3(Union *un,int p,int q){
    int pRoot = unionFind3(un,p);
    int qRoot = unionFind3(un,q);

    if(pRoot == qRoot){
        return ;
    }
    else if(un->rank[pRoot] < un->rank[qRoot]){
        un->parent[pRoot] = qRoot;
    }
    else if(un->rank[pRoot] > un->rank[qRoot]){
        un->parent[qRoot] = pRoot;
    }
    else{
        un->parent[pRoot] = qRoot;
        un->rank[qRoot] += 1;
    }
}