LeetCode每日一题——952. 按公因数计算最大组件大小

MaTF_

已于 2023-12-14 15:36:16 修改

阅读量1.3k

点赞数 1

分类专栏： LeetCode每日一题文章标签： leetcode 算法职场和发展

于 2022-07-30 18:23:31 首次发布

本文链接：https://blog.csdn.net/MaTF_/article/details/126073364

版权

LeetCode每日一题专栏收录该内容

6 篇文章 0 订阅

订阅专栏

参考资料：

https://leetcode.cn/problems/largest-component-size-by-common-factor/solution/an-gong-yin-shu-ji-suan-zui-da-zu-jian-d-amdx/

题目描述

题目传送门
在这里插入图片描述

样例

在这里插入图片描述

思路

本题是对并查集的简单应用，如果j能被i整除，则i、j、j/i应该被merge到一个集合里。

代码

#define maxn 20010

int sum[100005];
int f[100005];
int Rank[100005];

void init(int n){
    for(int i=0;i<=n;i++){
        f[i] = i;
        Rank[i] = 1;
    }
}
int find(int x){
    return (x==f[x])? x: (f[x]=find(f[x]));
}
void merge(int x, int y){
    int X=find(x), Y=find(y);
    if(X==Y) return;
    else if(Rank[X]>Rank[Y]) f[Y] = X;
    else if(Rank[Y]>Rank[X]) f[X] = Y;
    else{
        f[X] = Y;
        Rank[Y]++;
    }
}

class Solution {
public:
    int largestComponentSize(vector<int>& nums) {
        memset(sum, 0, sizeof(sum));
        memset(f, 0, sizeof(f));
        memset(Rank, 0, sizeof(Rank));
        int M = 0;
        for(int i=0;i<nums.size();i++)
            M = max(M, nums[i]);
        init(M);
        for(int i=0;i<nums.size();i++){
            for(int j=2;j*j<=nums[i];j++){
                if(nums[i]%j==0){
                    merge(nums[i], j);
                    merge(nums[i], nums[i]/j);
                }
            }
        }
        int ans = 0;
        for(int i=0;i<nums.size();i++){
            sum[find(nums[i])]++;
            ans = max(ans, sum[find(nums[i])]);
        }
        return ans;
    }
};