并查集模板

int UFFind(int *ufRoot, int a)
{
    if (ufRoot[a] < 0) {
        return a;
    }
    ufRoot[a] = UFFind(ufRoot, ufRoot[a]);	// 路径压缩
    return ufRoot[a];
}
void UFUnion(int *ufRoot, int a, int b) 
{
    int rtA = UFFind(ufRoot, a);
    int rtB = UFFind(ufRoot, b);
    if (rtA == rtB) {
        return;
    }
    if (ufRoot[rtA] < ufRoot[rtB]) {	// 按秩合并
        ufRoot[rtB] = rtA;
    } else if (ufRoot[rtA] > ufRoot[rtB]) {
        ufRoot[rtA] = rtB;
    } else {
        ufRoot[rtB] = rtA;		// 高度相同, B 挂到 A 上
        ufRoot[rtA]--;			// 更新 A 的高度
    }
}
int countComponents(int n, int** edges, int edgesSize, int* edgesColSize){
    int col = edgesColSize[0];
    int *ufRoot = (int *)malloc(sizeof(int) * n);
    for (int i = 0; i < n; i++) {
        ufRoot[i] = -1;			// 高度全部初始化为 -1 
    }
    for (int i = 0; i < edgesSize; i++) {
        UFUnion(ufRoot, edges[i][0], edges[i][1]);
    }
    int relt = 0; 
    for (int i = 0; i < n; i++) {
        if (ufRoot[i] < 0) {
            relt++;
        }
    }
    return relt;
}

使用场景:
LeetCode 323. 无向图中连通分量的数目

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值