tag: python3 、 数组
给定一个整数数组 a
,其中1 ≤ a[i] ≤ n
(n
为数组长度), 其中有些元素出现两次而其他元素出现一次。
找到所有出现两次的元素。
你可以不用到任何额外空间并在O(n)
时间复杂度内解决这个问题吗?
输入:
[4,3,2,7,8,2,3,1]
输出:
[2,3]
与找到所有数组中消失的数字类似的,要在 O(n)
的时间内且不用额外空间完成。那么我就可以通过遍历数组,遇到数字 i
则将 nums[i-1]
变为负数(因为出现的数字从一开始),如果对应的位置的数已经为负数,那么说明该数出现了第二次。
class Solution:
def findDuplicates(self, nums: List[int]) -> List[int]:
res = []
for i in nums:
if nums[abs(i)-1] < 0:
res.append(abs(i))
else:
nums[abs(i)-1] *= -1
return res
最终结果,运行时间432ms,超过83.74%;占用内存21.2MB,超过100.00%。