萌新练习写代码的每日一练

给你一个整数数组 nums 。

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

返回好数对的数目。

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

思路:先设置一个变量temp并赋值为0,使用双循环,然后if设置两个条件,符合条件的情况temp+1

代码:

class Solution:
    def numIdenticalPairs(self, nums: List[int]) -> int:
        temp = 0
        for i in range(len(nums)):
            for j in range(len(nums)):
                if nums[j] == nums[i] and i < j:
                    temp = temp + 1
        return temp

改进:因为条件之一是i<j,所以可以在第二层循环那里加上范围,后边if只需要跟一个条件即可

改进代码:for j in range(i+1,len(nums))

学到的新思路:做一个字典,字典的key为数字(nums),value为数字出现的次数。遍历字典的值,计算好数对个数。
好数对计算方法:出现一次的数字没有好数对,出现两次的数字拥有1个好数对,出现三次的拥有2+1个好数对,1(号数字)和2,2和3,1和3。以此类推出现n次的数字拥有1+2+3+...+(n-1)个好数对,由等差数列化简后得n*(n-1)/2

代码:

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

题目来自LeetCode1512

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值