Leetcode practice (10)

Tuple with Same Product

• 题目描述：给定一个无重复且为正整数的数组，返回一个四维元组(a, b, c, d), 元组内部满足 a * b = c * d 且 a != b != c != d, 求满足这样的元组个数
• 思路：最简单的思路就是暴力求解，嵌套遍历数组4次，将符合条件的组合计数即可，但是发现这样复杂度是 O(n^4), 使用c++也超时了，因此要找到一种更优的解法。先从条件 a * b = c * d下手，数组中两个元素的乘积，其实就是从数组中选取两个进行组合，即 C(2, n)，这些组合之间的乘积，即为目标条件，所以下面需要知道哪些组合可以得到相同的乘积，将这些组合数记录下来，之后相同乘积下的组合数再取 C(2, n), 即得到目标统计结果，这里注意的是 需要对C(2, n) * 8, 一个四维元组有固定8种组合。
• code：

// 暴力求解
class Solution {
public:
    int tupleSameProduct(vector<int>& nums) {
        int count = 0;
        for (int i = 0; i < nums.size(); i++){
            for (int j = 0;j < nums.size(); j++){
                for (int k = 0; k < nums.size(); k++) {
                    for (int l = 0; l < nums.size(); l++){
                        if (nums[i] != nums[j] && nums[i] != nums[k] && nums[i] != nums[l] && nums[j] != nums[k] && nums[j] != nums[l] && nums[l] != nums[k]) {
                            if (nums[i] * nums[j] == nums[l] * nums[k]){
                                count += 1;
                                }
                        }
                        
                    }
                }
            }
        }
        return count;
    }
};

# 利用hashtable求解
class Solution:
    def tupleSameProduct(self, nums: List[int]) -> int:
        d = {}
        
        for i in range(len(nums)):
            for j in range(i + 1, len(nums)):
                if nums[i] * nums[j] in d:
                    d[nums[i] * nums[j]] += 1
                else:
                    d[nums[i] * nums[j]] = 1
                
        count = 0
        
        for _, v in d.items():
            count += ((v * (v-1)) // 2) * 8

        return count

• 总结：一直想着怎么最优遍历组合，还想到先排序再怎么查找，陷入思维的误区坑，因为4个元素散落在数组当中，任意4个元素都有机会满足条件，所以四个元素均可变，如果固定一个，改动其他就是暴力解法，但是暴力解法时间复杂度过高。其实，数组给定了，数组能组合哪些乘积也固定了，因此之需要把乘积作为key，记录有哪些组合满足条件即可，完全不需要关心具体组合是什么了。还是刷的题量不够，对于题目的思路解法存在太多干扰信息。