Leetcode C++ 《第181场周赛-2》 1390. 四因数

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;
    }
};