leetcode-952:按公因数计算最大组件大小

简介: leetcode-952:按公因数计算最大组件大小

题目

题目连接

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

有 nums.length 个节点,按从 nums[0] 到 nums[nums.length - 1] 标记;

只有当 nums[i] 和 nums[j] 共用一个大于 1 的公因数时,nums[i] 和 nums[j]之间才有一条边。

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

示例 1:

输入:nums = [4,6,15,35]
输出:4

示例 2:

输入:nums = [20,50,9,63]
输出:2

示例 3:

输入:nums = [2,3,6,7,4,12,21,39]
输出:8

解题

方法一:带权重有向图+并查集(超时)

带权重的图+并查集

写法一:

每个节点还带了权重weight[i],表示直接指向当前节点的 节点个数

最后只需要找到权重最大的,就是我们想要的答案。

能直接遍历weight来获得最大的集合元素个数?显然不能直接这么做,还需要对他进行路径压缩,让元素直接指向根节点。

class UnionFind{
private:
    vector<int> parent;
    vector<int> weight;
public:
    UnionFind(int n){
        parent.resize(n);
        weight.resize(n);
        for(int i=0;i<n;i++){
            parent[i]=i;
            weight[i]=1;//直接指向当前节点的数量(包含自身,注意孙子节点,不算直接指向)
        }
    }
    int find(int index){
        if(parent[index]==index) return index;
        int origin=parent[index];
        parent[index]=find(parent[index]);
        weight[parent[index]]++;//指向根节点的节点数量+1
        weight[origin]--;//指向原先节点的节点数量-1
        return parent[index];
    }
    void unite(int index1,int index2){
        int p1=find(index1);
        int p2=find(index2);
        if(p1!=p2){
            parent[p1]=p2;
            weight[p2]++;//指向根节点的节点数量+1
        }
    }
    int findMaxSet(){
        for(int i=0;i<weight.size();i++){
            find(i);
        }
        int res=0;
        for(int i=0;i<weight.size();i++){
            res=max(res,weight[i]);
        }
        return res;
    }
};
class Solution {
public:
    int largestComponentSize(vector<int>& nums) {
        int n=nums.size();
        UnionFind uf(n);
        unordered_map<int,int> mp;
        int id=0;
        for(int i=0;i<n;i++){
            if(!mp.count(nums[i])){
                mp[nums[i]]=id++;
            }
        }
        for(int i=0;i<n;i++){
            for(int j=i+1;j<n;j++){
                int id1=mp[nums[i]];
                int id2=mp[nums[j]];
                if(gcd(nums[i],nums[j])>1){
                    uf.unite(id1,id2);
                }
            }
        }
        return uf.findMaxSet();
    }
};

写法二:

weight表示,指向当前节点的节点数(包含孙子节点等)

只要在unite里面加上另一个子树的所有节点 数就行了,find过程中进行的路径压缩,只是调整了指向根节点,但是该根节点的所有节点数是不会变的。

class UnionFind{
private:
    vector<int> parent;
    vector<int> weight;
public:
    UnionFind(int n){
        parent.resize(n);
        weight.resize(n);
        for(int i=0;i<n;i++){
            parent[i]=i;
            weight[i]=1;//指向当前节点的数量(包含孙子节点等)
        }
    }
    int find(int index){
        if(parent[index]==index) return index;
        parent[index]=find(parent[index]);
        return parent[index];
    }
    void unite(int index1,int index2){
        int p1=find(index1);
        int p2=find(index2);
        if(p1!=p2){
            parent[p1]=p2;
            weight[p2]+=weight[p1];//指向根节点的节点数量 加上 子树的节点数量
        }
    }
    int findMaxSet(){
        for(int i=0;i<weight.size();i++){
            find(i);
        }
        int res=0;
        for(int i=0;i<weight.size();i++){
            res=max(res,weight[i]);
        }
        return res;
    }
};
class Solution {
public:
    int largestComponentSize(vector<int>& nums) {
        int n=nums.size();
        UnionFind uf(n);
        unordered_map<int,int> mp;
        int id=0;
        for(int i=0;i<n;i++){
            if(!mp.count(nums[i])){
                mp[nums[i]]=id++;
            }
        }
        for(int i=0;i<n;i++){
            for(int j=i+1;j<n;j++){
                int id1=mp[nums[i]];
                int id2=mp[nums[j]];
                if(gcd(nums[i],nums[j])>1){
                    uf.unite(id1,id2);
                }
            }
        }
        return uf.findMaxSet();
    }
};

主要超时原因还是因为两层for循环

方法二:枚举质因数+并查集

参考链接

基本思路:把关联的加入到并查集中,最后遍历nums,相同的根节点那么数量+1,这样就能找到最多元素的集合了。

由于nums.length最大为2 ∗ 1 0 4 2*10^42104lianjie

遍历每个值,两两匹配会超时。

因此采用枚举公因式的方式,如果k>=2,且num%k==0,那么k为公约数,对num和k进行联结,以及num和num/k进行联结。不管k或者num/k有没有在nums中这都不要紧,因为后面遍历nums的时候不会遍历到它们。

相比O ( 4 ∗ 1 0 8 ) O(4*10^8)O(4108)的复杂度来的更小。

class UnionFind{
private:
    vector<int> parent;
public:
    UnionFind(int n){
        parent.resize(n);
        for(int i=0;i<n;i++){
            parent[i]=i;
        }
    }
    int find(int index){
        if(parent[index]==index) return index;
        return parent[index]=find(parent[index]);
    }
    void unite(int index1,int index2){
        int p1=find(index1);
        int p2=find(index2);
        if(p1==p2) return;
        parent[p1]=p2;
    }
};
class Solution {
public:
    int largestComponentSize(vector<int>& nums) {
        int maxLength=*max_element(nums.begin(),nums.end());
        int n=nums.size();
        UnionFind uf(maxLength+1);
        for(int num:nums){
            for(int k=sqrt(num);k>=2;k--){
                if(num%k==0){
                    uf.unite(num,k);//不管k属不属于num都没有关系,把他添加到集合中。因为后续还要遍历一遍nums
                    uf.unite(num,num/k);
                }
            }
        }
        vector<int> cnt(maxLength+1);
        int res=0;
        for(int num:nums){//遍历nums中的值,同一集合的+1
            cnt[uf.find(num)]++;
            res=max(res,cnt[uf.find(num)]);
        }
        return res;
    }
};
相关文章
|
6月前
|
存储
LeetCode刷题---817. 链表组件(哈希表)
LeetCode刷题---817. 链表组件(哈希表)
【Leetcode -817.链表组件 -1019.链表中的下一个更大节点】
【Leetcode -817.链表组件 -1019.链表中的下一个更大节点】
47 0
|
6月前
leetcode-817:链表组件
leetcode-817:链表组件
41 0
LeetCode每日一题——817. 链表组件
给定链表头结点 head,该链表上的每个结点都有一个 唯一的整型值 。同时给定列表 nums,该列表是上述链表中整型值的一个子集。
95 0
LeetCode每日一题——817. 链表组件
|
算法
【DAY29】LeetCode力扣算法题 [817. 链表组件 ]
学习LeetCode力扣算法题 [817. 链表组件 ]。
187 0
【DAY29】LeetCode力扣算法题 [817. 链表组件 ]
|
2月前
|
Unix Shell Linux
LeetCode刷题 Shell编程四则 | 194. 转置文件 192. 统计词频 193. 有效电话号码 195. 第十行
本文提供了几个Linux shell脚本编程问题的解决方案,包括转置文件内容、统计词频、验证有效电话号码和提取文件的第十行,每个问题都给出了至少一种实现方法。
LeetCode刷题 Shell编程四则 | 194. 转置文件 192. 统计词频 193. 有效电话号码 195. 第十行
|
3月前
|
Python
【Leetcode刷题Python】剑指 Offer 32 - III. 从上到下打印二叉树 III
本文介绍了两种Python实现方法,用于按照之字形顺序打印二叉树的层次遍历结果,实现了在奇数层正序、偶数层反序打印节点的功能。
56 6
|
3月前
|
搜索推荐 索引 Python
【Leetcode刷题Python】牛客. 数组中未出现的最小正整数
本文介绍了牛客网题目"数组中未出现的最小正整数"的解法,提供了一种满足O(n)时间复杂度和O(1)空间复杂度要求的原地排序算法,并给出了Python实现代码。
113 2
|
19天前
|
机器学习/深度学习 人工智能 自然语言处理
280页PDF,全方位评估OpenAI o1,Leetcode刷题准确率竟这么高
【10月更文挑战第24天】近年来,OpenAI的o1模型在大型语言模型(LLMs)中脱颖而出,展现出卓越的推理能力和知识整合能力。基于Transformer架构,o1模型采用了链式思维和强化学习等先进技术,显著提升了其在编程竞赛、医学影像报告生成、数学问题解决、自然语言推理和芯片设计等领域的表现。本文将全面评估o1模型的性能及其对AI研究和应用的潜在影响。
16 1
|
2月前
|
数据采集 负载均衡 安全
LeetCode刷题 多线程编程九则 | 1188. 设计有限阻塞队列 1242. 多线程网页爬虫 1279. 红绿灯路口
本文提供了多个多线程编程问题的解决方案,包括设计有限阻塞队列、多线程网页爬虫、红绿灯路口等,每个问题都给出了至少一种实现方法,涵盖了互斥锁、条件变量、信号量等线程同步机制的使用。
LeetCode刷题 多线程编程九则 | 1188. 设计有限阻塞队列 1242. 多线程网页爬虫 1279. 红绿灯路口