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