Leetcode C++ 《第181场周赛-2》 1390. 四因数
1. 题目
给你一个整数数组 nums,请你返回该数组中恰有四个因数的这些整数的各因数之和。
如果数组中不存在满足题意的整数,则返回 0 。
示例:
输入:nums = [21,4,7]
输出:32
解释:
21 有 4 个因数:1, 3, 7, 21
4 有 3 个因数:1, 2, 4
7 有 2 个因数:1, 7
答案仅为 21 的所有因数的和。
提示:
1 <= nums.length <= 10^4
1 <= nums[i] <= 10^5
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/four-divisors
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. 思路
- 看别人的代码,看到一个巧妙的解法,使用加法的思路,最大的复杂度是100000*100000
- 一个用来记录因数个数,一个用来记录因数和
- 数组要记得初始化
3. 代码
class Solution {
public:
int sumFourDivisors(vector<int>& nums) {
int factorNums[100001];
int sums[100001];
int n = 1;
for (int i = 0; i < nums.size(); i++)
n = max(n, nums[i]);
for (int i = 1; i <= n; i++) {
sums[i] = 0;
factorNums[i] = 0;
}
for (int i = 1; i <= n; i++) {
for (int j = i; j <= n; j = j+i) {
factorNums[j]++;
sums[j] += i; //i作为因数
}
}
int res = 0;
for (int i = 0; i < nums.size(); i++) {
if (factorNums[nums[i]] == 4)
res += sums[nums[i]];
}
return res;
}
};