leetcode.442 数组中重复的数据(python3)

给定一个整数数组 a,其中1 ≤ a[i] ≤ n (n为数组长度), 其中有些元素出现两次而其他元素出现一次。

找到所有出现两次的元素。

你可以不用到任何额外空间并在O(n)时间复杂度内解决这个问题吗?

示例:

输入:
[4,3,2,7,8,2,3,1]

输出:
[2,3]


根据题意要求O(n)时间复杂度内,那么此题首先想到的就是暴力破解,暴力破解只需要循环一次就OK了。
代码如下:

class Solution:
    def findDuplicates(self, nums: 'List[int]') -> 'List[int]':
 
        if len(nums) < 2:
            return []

        nums.sort()    
        bool_temp = nums[-1]==nums[-2]

        i = 0
        while i < len(nums)-1:
            if nums[i] == nums[i+1]:
                nums.pop(i)
                i += 1
            else:
                nums.pop(i)
        if not bool_temp:
            nums.pop(-1)
            
        return nums

虽然过了,但是跑出来的结果1000多ms,太慢了,然后就想其他办法,想到了set的去重功能,那么变量去重后的,是不是更省时间呢?

class Solution:
    def findDuplicates(self, nums: 'List[int]') -> 'List[int]':
    
        temp_nums = list(set(nums))           # 去重
         
        return [i for i in temp_nums if nums.count(i)==2]	# 方法1
    # 或者
        for i in temp_nums:				# 方法2
            nums.remove(i)
        return nums

遗憾的是两种方法全部超时,好尴尬,不知道为什么这种遍历的元素少的反而超时了。那么只有再继续想办法。不知道大家有没有了解过collections库,其中的Counter方法很不错,然后就生成了以下代码:

class Solution:
    def findDuplicates(self, nums: 'List[int]') -> 'List[int]':
        re = []
        import collections
        obj = collections.Counter(nums)
        for i in obj:
            if obj[i] == 2:
                re.append(i)
        
        return re

180ms,终于能进前面的5%了

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值