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

头文件 UnionFind2.h

#ifndef UNIONFIND2_H_INCLUDED
#define UNIONFIND2_H_INCLUDED
#include "stdlib.h"
#include "ASSERT.h"
typedef struct{
    int* parent;
    int count;
}Union;

void unionFind2Init(Union *un,int n);
void freeUnion2(Union *un);
int unionFind2(Union *un,int p);
int isConnected2(Union *un,int p,int q);
void unionElements2(Union *un,int p,int q);
#endif // UNIONFIND_H_INCLUDED

c代码 UnionFind2.c

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

/*********************************************
*函数名：返回并查集中p的ID
*********************************************/
int unionFind2(Union *un,int p){
    assert(p >= 0 && p< un->count);
    while(p != un->parent[p]){
        p = un->parent[p];
    }
    return p;
}
/*********************************************
*函数名：判断并查集中p和q是否是一个集合
*********************************************/
int isConnected2(Union *un,int p,int q){
    return unionFind2(un,p) == unionFind2(un,q) ;
}
/*********************************************
*函数名：并查集的合并
*********************************************/
void unionElements2(Union *un,int p,int q){

    int pRoot = unionFind2(un,p);
    int qRoot = unionFind2(un,q);

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