leetcode_1512 好数对的数目

96 阅读1分钟

要求

给你一个整数数组 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

提示:

  • 1 <= nums.length <= 100
  • 1 <= nums[i] <= 100

核心代码

class Solution:
    def numIdenticalPairs(self, nums: List[int]) -> int:
        from collections import Counter
        dic = Counter(nums)
        return sum((value - 1) * value // 2 for value in dic.values())

image.png

解题思路:统计每个数出现的频率,如果一个数出现了N次,那么它对应的好数对就有组合数Cn2C_n^2个,之所以是组合,是因为我们有个限制是i < j。时间复杂度:O(N)空间复杂度:O(N)