给定一个整数数组 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%了