2470. 最小公倍数为 K 的子数组数目

题目描述:

给你一个整数数组 nums 和一个整数 k ,请你统计并返回 nums 的 子数组 中满足 元素最小公倍数为 k 的子数组数目。

子数组 是数组中一个连续非空的元素序列。

数组的最小公倍数 是可被所有数组元素整除的最小正整数。

题目来源

示例1.

输入:nums = [3,6,2,7,1], k = 6
输出:4
解释:以 6 为最小公倍数的子数组是:
- [3,6]
- [3,6,2]
- [6]
- [6,2]

示例2.

输入:nums = [3], k = 2
输出:0
解释:不存在以 2 为最小公倍数的子数组。

解题思路:

此题主要需要解决两个问题:

1. 求两个或多个数的最小公倍数

2.如何求子数组

a和b的最小公倍数求法:a*b/最大公约数

最大公约数求法:辗转相除法,当a/b有余数时,令a=b,b=余数,继续相除直到a/b没有余数

代码:

class Solution {
public:
    int GCD(int a, int b){
        if(b==0)
        {
            return 1;
        }
        else if(a%b==0)
        {
            return b;
        }
        else
            return GCD(b, a%b);
    }
    int LCM(int a, int b)
    {
        int gcd = GCD(a, b);
        return a*b/gcd;
    }
    int subarrayLCM(vector<int>& nums, int k) {
        
        int m=0;
        int sum=0;
        for(int i=0; i<nums.size(); ++i)
        {
            m = nums[i];
            for(int j=i; j<nums.size(); ++j)
            {
                m = LCM(m, nums[j]);
                if(m==k)
                {
                    sum++;
                }      
                else if(m>k)
                {
                    break;
                }        
            }   
        }
        return sum;
    }
};

求子数组的方法是使用两层循环,第一层循环首先确定了子数组的首元素,并设置首元素为最小公倍数,第二层判断当前元素与当前最小公倍数最小公倍数是否等于k,等于则表明此数与前面的数组成的子数组符合条件,子数组个数加1.

两层循环的初始条件位置相同,是为了表示当该元素自己本身就等于k时,自己可以组成一个子数组。else if的判断,表示当此次求得的最小公倍数大于k时,该数与前面的数的最小公倍数已经不符合等于k的题意,直接提前结束循环。而如果m<k,例如m=3, k=6,m自身的最小公倍数就是3,然而这并不能代表后面没有最小公倍数为6的数,所以还要继续向后判断。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
首先,我们需要知道一个结论:一个数的质因数目等于其所有质因的指数之和加1。比如,$24=2^3\times3$,其质因数目为$3+1=4$。 接下来,我们可以使用暴力枚举数组的方法,对每个数组求出其最小公倍数,然后计算其质因数目之和即可。时间复杂度为$O(n^3)$,其中$n$为数组长度,由于数组数量为$O(n^2)$,对每个数组最小公倍数的时间复杂度为$O(n)$。 代码如下: ```python def gcd(a, b): if b == 0: return a return gcd(b, a % b) def lcm(a, b): return a * b // gcd(a, b) def count_factors(num): count = 0 for i in range(2, int(num**0.5)+1): while num % i == 0: count += 1 num //= i if num > 1: count += 1 return count def subarray_lcm_factor_sum(arr): n = len(arr) ans = 0 for i in range(n): for j in range(i, n): subarr_lcm = arr[i] for k in range(i+1, j+1): subarr_lcm = lcm(subarr_lcm, arr[k]) ans += count_factors(subarr_lcm) return ans ``` 然而,上述方法的时间复杂度较高,不能通过本题。考虑优化。 观察题目的数据范围,发现数组中每个元素都小于$10^6$,因此一个数组最小公倍数也不可能超过$10^6$,这启示我们可以对每个质数预处理出其在$[1,10^6]$范围内的倍数中出现的次数。具体地,我们可以使用线性筛法预处理出所有小于等于$10^6$的质数,然后对每个质数$p$,预处理出其在$[1,10^6]$范围内的倍数中出现的次数$cnt_p$,即: $$cnt_p=\sum_{i=1}^{\lfloor 10^6/p\rfloor}\lfloor\frac{i}{p}\rfloor$$ 这里的$\lfloor x\rfloor$表示不超过$x$的最大整数。例如,当$p=2$时,有$cnt_2=\lfloor 10^6/2\rfloor+\lfloor 10^6/4\rfloor+\lfloor 10^6/8\rfloor+\cdots=500000+250000+125000+\cdots$。这个式可以用等比数列求和公式简化为$cnt_2=10^6-1$。 预处理完成后,对于每个数组,我们可以依次枚举$[1,10^6]$范围内的质数$p$,统计其在数组最小公倍数中出现的次数。具体地,设当前枚举到的质数为$p$,统计$p^{cnt_p}$在数组最小公倍数中出现的次数。这个数量等于数组中所有元素中$p$的指数的最小值乘以$cnt_p$。 具体地,我们可以用一个长度为$10^6+1$的数组$factor$来存储每个数对应的最小质因。我们可以先用线性筛法求出所有小于等于$10^6$的质数,并将它们的倍数都标记上最小质因。然后,对于每个数组,我们可以使用一个长度为$10^6+1$的数组$cnt$来统计每个质数的指数之和,然后根据上述公式计算质因数目之和。 总时间复杂度为$O(n\log\log 10^6)$,其中$n$为数组长度。 代码如下: ```python def subarray_lcm_factor_sum(arr): n = len(arr) # 预处理每个数的最小质因 factor = [0] * (10**6+1) primes = [] for i in range(2, 10**6+1): if factor[i] == 0: factor[i] = i primes.append(i) for p in primes: if p > factor[i] or i*p > 10**6: break factor[i*p] = p # 预处理每个质数在[1,10^6]范围内的倍数中出现的次数 cnt = [0] * (10**6+1) for p in primes: cnt[p] = 10**6 // p for i in range(2, cnt[p]+1): cnt[p] -= cnt[i*p] # 枚举数组,统计质因数目之和 ans = 0 for i in range(n): for j in range(i, n): subarr_lcm = arr[i] for k in range(i+1, j+1): subarr_lcm = lcm(subarr_lcm, arr[k]) for p in primes: if p > subarr_lcm: break exp_sum = float('inf') for x in arr[i:j+1]: exp_sum = min(exp_sum, get_exponent(x, p)) ans += exp_sum * cnt[p] return ans ``` 其中,$get\_exponent(x,p)$用于计算$x$在质因数分解中$p$的指数,代码如下: ```python def get_exponent(num, prime): exp = 0 while num % prime == 0: exp += 1 num //= prime return exp ```

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值