1512. 好数对的数目

最新推荐文章于 2024-09-18 15:03:01 发布

yungezier

最新推荐文章于 2024-09-18 15:03:01 发布

阅读量103

点赞数

文章标签： leetcode

本文链接：https://blog.csdn.net/yungezier/article/details/109173666

版权

1512. 好数对的数目

给你一个整数数组 nums 。

如果一组数字 $(i, j)$ 满足 $n u m s [i] = = n u m s [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:
        n = len(nums)
        count =0
        for i in range(n-1):
            for j in range(i+1,n):
                if nums[i] == nums[j]:
                    count += 1
        return count

二、哈希表+数学规律

class Solution:
    def numIdenticalPairs(self, nums: List[int]) -> int:
        hmap ={}
        count = 0
        for i in nums:
            if i in hmap:
                hmap[i] +=1
            else: hmap[i] =1
        
        for j in hmap.values():
            count += (j*(j-1)//2)
            
        return count