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

1726.同积元组

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

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

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

如果对您有帮助,请点赞关注支持我,谢谢!❤
如有错误或者不足之处,敬请指正!❤