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