leetcode--python--1512

1512. 好数对的数目

给你一个整数数组 nums 。

如果一组数字 (i,j) 满足 nums[i] == nums[j] 且 i < j ,就可以认为这是一组 好数对 。

返回好数对的数目。

class Solution(object):
    def numIdenticalPairs(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        unique = set(nums)
        result = 0
        for i in unique:
            n = nums.count(i)
            result += n*(n-1)/2
        return(result)

leetcode--python--1512_第1张图片
代码思路,我们只需要在数组中通过set函数得到不重复的数字有哪些,好数对的数目即为{[(n-1)+1]/2}*(n-1),也就是n(n-1)/2

你可能感兴趣的:(leetcode-python)