算法|每日一题|同积元组|哈希统计

星不易2023-10-20 10:01

1726.同积元组

原题地址：力扣每日一题：同积元组

给你一个由不同正整数组成的数组 nums ，请你返回满足 a * b = c * d 的元组 (a, b, c, d) 的数量。其中 a、b、c 和 d 都是 nums 中的元素，且 a != b != c != d 。

java 复制代码

class Solution {
    public int tupleSameProduct(int[] nums) {
        int n = nums.length;
        Map<Integer, Integer> cnt = new HashMap<>();
        for (int i = 0; i < n; i++) {
            for (int j = i + 1; j < n; j++) {
            	// 存储两个数的积
                int key = nums[i] * nums[j];
                // 如果有该乘积了就加1，如果没有就赋值1
                cnt.put(key, cnt.getOrDefault(key, 0) + 1);
            }
        }
        int ans = 0;
        for (Integer v : cnt.values()) {
        	// v是每个乘积的次数，超过1次说明有符合题目条件的元组
        	// 对于得到相同乘积的不同个元组（a，b），总共能再组合v*(v-1)/2次
        	// 每种包含四个元素的组合有八种排列方式
            ans += v * (v - 1) * 4;
        }
        return ans;
    }
}

如果对您有帮助，请点赞关注支持我，谢谢！❤

如有错误或者不足之处，敬请指正！❤