leetcode--好数对的数目

 题目是LeetCode第197场周赛的第一题，链接：1512. 好数对的数目。具体描述为：给你一个整数数组nums。如果一组数字(i,j)满足 nums[i] == nums[j]且i < j，就可以认为这是一组好数对。返回好数对的数目。

 示例1:

输入：nums = [1,2,3,1,1,3]
输出：4
解释：有 4 组好数对，分别是 (0,3), (0,4), (3,4), (2,5) ，下标从 0 开始

 示例2:

输入：nums = [1,1,1,1]
输出：6
解释：数组中的每组数字都是好数对

 示例3:

输入：nums = [1,2,3]
输出：0

 没啥难度，直接统计各个数字的出现次数count[num]，那么最终数字num的好数对数量就是count[num]*(count[num]-1)/2。时间复杂度为$O(n)$，空间复杂度为$O(n)$。

 JAVA版代码如下：

class Solution {
    public int numIdenticalPairs(int[] nums) {
        Map<Integer, Integer> count = new HashMap<>();
        for (int num : nums) {
            count.put(num, count.getOrDefault(num, 0) + 1);
        }
        int res = 0;
        for (int c : count.values()) {
            res += c * (c - 1) / 2;
        }
        return res;
    }
}

 提交结果如下:

 Python版代码如下：

class Solution:
    def numIdenticalPairs(self, nums: List[int]) -> int:
        num2count = collections.defaultdict(int)
        for num in nums:
            num2count[num] += 1
        result = 0
        for num in num2count:
            result += (num2count[num] * (num2count[num] - 1)) // 2
        return result

 提交结果如下: