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