【算法题】2521. 数组乘积中的不同质因数数目
题目:
给你一个正整数数组 nums ,对 nums 所有元素求积之后,找出并返回乘积中 不同质因数 的数目。
注意:
质数 是指大于 1 且仅能被 1 及自身整除的数字。
如果 val2 / val1 是一个整数,则整数 val1 是另一个整数 val2 的一个因数。
示例 1:
输入:nums = [2,4,3,7,10,6]
输出:4
解释:
nums 中所有元素的乘积是:2 * 4 * 3 * 7 * 10 * 6 = 10080 = 25 * 32 * 5 * 7 。
共有 4 个不同的质因数,所以返回 4 。
示例 2:
输入:nums = [2,4,8,16]
输出:1
解释:
nums 中所有元素的乘积是:2 * 4 * 8 * 16 = 1024 = 210 。
共有 1 个不同的质因数,所以返回 1 。
提示:
1 <= nums.length <= 10^4
2 <= nums[i] <= 1000
java 代码:
class Solution {public int distinctPrimeFactors(int[] nums) {// 不同的质数集合HashSet<Integer> primeSet = new HashSet<>();// 不同的质因数集合HashSet<Integer> PrimeFactorSet = new HashSet<>();// 不计算乘积,直接遍历数组for (int n : nums) {if (PrimeFactorSet.contains(n)) {// 重复continue;}if (isPrime(n, primeSet)) {PrimeFactorSet.add(n);}// 枚举,直到一半开根号for (int i = 2; i <= Math.sqrt(n); i++) {// 判断i是否质数if (isPrime(i, primeSet)) {// i是质数,判断是否为n的质因数while (n % i == 0) {PrimeFactorSet.add(i);// 查找下一个n /= i;// 判断是否分解完全if (isPrime(n, primeSet)) {PrimeFactorSet.add(n);break;}}}}}return PrimeFactorSet.size();}// 判断是否质数private boolean isPrime(int n, HashSet<Integer> primeSet) {if (primeSet.contains(n)) {return true;}// 枚举,直到一半开根号for (int i = 2; i <= Math.sqrt(n); i++) {if (n % i == 0) {return false;}}// 缓存质数primeSet.add(n);return true;}
}