给你一个整数数组 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
题解:
审题!分析题目让我们实现什么,所谓好数对,即前面出现过的元素后面又出现了。因此,分析思路如下:
首先,可以对输入列表进行检查,判断是否存在重复元素,如果不存在重复元素,那么是不是这个数组列表就不存在所谓的好数对了,直接return 0 即可。
那么,在存在好数对的前提下,我们只需要循环遍历两次数组,即可找到我们需要的好数对。
方法一:
def numIdenticalPairs(self, nums: List[int]) -> int:
if len(set(nums)) == len(nums):
return 0
else:
sum = 0
for i in range(0,len(nums)):
for j in range(i+1,len(nums)):
if nums[i] == nums[j]:
sum += 1
return sum
这种方法就是比较传统的笨方法,在固定一个数据的前提下,循环遍历后面的数据,并判断是否相等。
方法二:
def numIdenticalPairs(self, nums: List[int]) -> int:
if len(set(nums)) == len(nums):
return 0
else:
return sum(nums[i+1:].count(num) for i, num in enumerate(nums))
这种思路就相对高级一些, 省去了许多变量的引用,直接调用enumerate函数,计算从i+1后,num在nums里出现的次数,就可计算出所有好数对。nice~
今天周末,回家老婆做了好吃的咸蛋黄茄子和魔芋烧鸭,白天工作也比较顺利,学到了蛮多东西,开心开心,明天看看更一篇QT的笔记。
周末愉快~