并查集应用

并查集

至于并查集的教程,可以参考文章算法学习笔记(1) : 并查集,写的很详细和通俗易懂,本文就不再介绍。下面给出并查集代码:

class UnionFind{
private:
    vector<int> parent;
    vector<int> rank;  // 合并两个不同元素的parent参考依据
public:
    UnionFind(int n){
        parent = vector<int>(n);
        rank = vector<int>(n);
        for(int i = 0; i < n ; i++){
            parent[i] = i;
        }
    }
    void uni(int x, int y){
        int rootx = find(x);
        int rooty = find(y);
        if(rootx != rooty){
            if(rank[rootx] > rank[rooty]){
                parent[rooty] = rootx;
            }else if(rank[rootx] < rank[rooty]){
                parent[rootx] = rooty;
            }else{
                parent[rooty] = rootx;
                rank[rootx]++;
            }
        }
    }
    int find(int x){
        if(parent[x] != x){
            parent[x] = find(parent[x]);
        }
        return parent[x];
    }
};
题目描述

给定一个由不同正整数的组成的非空数组 nums ,考虑下面的图:

  • nums.length 个节点,按从 nums[0]nums[nums.length - 1] 标记;
  • 只有当 nums[i]nums[j] 共用一个大于 1 的公因数时,nums[i]nums[j] 之间才有一条边。

返回 图中最大连通组件的大小 。

思路解析

返回图中最大连通组件的大小,换言之,找出图中最大子图的节点数目,可以使用并查集来对其进行求解,对于 nums 中的每一个元素 num ,遍历 [2, n u m \sqrt{num} num ],找出所有符合公因子的值,其作为合并的桥梁,最终将所有满足要求的元素 num 放到同一个集合中去。

说明
  • 并查集数组大小为数组 nums 中的最大值。
  • 对于范围 [ 2 , n u m ] [2, \sqrt{num}] [2,num ] 内的每个正整数 i i i,如果 i i i n u m num num 的因数,则 n u m num num i i i n u m i \frac{num}{i} inum 都属于同一个组件。
完整代码
#include <iostream>
#include <vector>
#include <bits/stdc++.h>
using namespace std;

class UnionFind{
private:
    vector<int> parent;
    vector<int> rank;
public:
    UnionFind(int n){
        parent = vector<int>(n);
        rank = vector<int>(n);
        for(int i = 0; i < n ; i++){
            parent[i] = i;
        }
    }
    void uni(int x, int y){
        int rootx = find(x);
        int rooty = find(y);
        if(rootx != rooty){
            if(rank[rootx] > rank[rooty]){
                parent[rooty] = rootx;
            }else if(rank[rootx] < rank[rooty]){
                parent[rootx] = rooty;
            }else{
                parent[rooty] = rootx;
                rank[rootx]++;
            }
        }
    }
    int find(int x){
        if(parent[x] != x){
            parent[x] = find(parent[x]);
        }
        return parent[x];
    }
};

class Solution {
public:
    int largestComponentSize(vector<int>& nums) {
        int m = *max_element(nums.begin(), nums.end()); // 找到vector的最大元素
        UnionFind uf(m + 1);
        for(int num : nums){
            for(int i = 2; i * i <= num; i++){
                if( num % i == 0){
                    uf.uni(num, i);
                    uf.uni(num, num / i);
                }
            }
        }
        vector<int> counts(m + 1);
        int ans = 0;
        for(int num : nums){
            int root = uf.find(num);
            counts[root]++;
            ans = max(ans, counts[root]);
        }
        return ans;
    }
};

int main(){
    // vector<int> nums = {4, 6, 15, 35};
    // vector<int> nums = {20, 50, 9, 63};
    vector<int> nums = {2, 3, 6, 7, 4, 12, 21, 39};

    Solution so;
    int ans = so.largestComponentSize(nums);
    cout<< ans << endl;

    system("pause");
    return 0;
}
// output: 8
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值