[leetcode]1390. 四因數

【題目】

給你一個整數數組 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

【思路】

暴力枚舉每一個數的約數（因數）。如果這個數的因數個數是4，計算這個數所有因數的和。優化的點，因爲因素的個數都是成對出現的，所以我們枚舉因素只需要枚舉一半（比較小的一部分,  從1 枚舉到 ）。

【代碼】

class Solution {
public:
    int sumFourDivisors(vector<int>& nums) {
        int res = 0;
        for(int i = 0 ; i < nums.size(); i++) {
            int sum = 0, cnt = 0;
            for(int j = 1 ; j * j <= nums[i]; j++ ) {
                if( nums[i] % j == 0 ) {
                    sum += j , cnt++;
                    if( j * j  != nums[i] ) sum += nums[i] / j, cnt++;
                }
            }
            if(cnt == 4) res += sum;
        }
        return res;
    }
};