[LeetCode] 442. 数组中重复的数据
题目链接: https://leetcode-cn.com/problems/find-all-duplicates-in-an-array
难度:中等
通过率:61.5%
题目描述:
给定一个整数数组 a,其中1 ≤ a[i] ≤ n ( n 为数组长度), 其中有些元素出现 两次 而其他元素出现 一次 。
找到所有出现 两次 的元素。
你可以不用到任何额外空间并在O( n )时间复杂度内解决这个问题吗?
示例:
**输入:**
[4,3,2,7,8,2,3,1]**输出:**
[2,3]
思路:
思路一:排序
通过索引号排序,比如数字4
放到索引3
的位置,最后找排序后数组,与索引号没有相差1
便是重复元素
思路二:绝对值
借用索引号,因为是在1~n
之间,那么我们可以用索引0
表示数字1
,索引1
表示数字2
...,当有个数字num
,我们将num - 1
的位置的数字取相反数,连续两次取相反数会变回来,便可判断元素出现次数。
所以时间复杂度为$O(n)$
相似题型:448. 找到所有数组中消失的数字
代码:
思路一:
class Solution:def findDuplicates(self, nums: List[int]) -> List[int]:res = []for i in range(len(nums)):while nums[nums[i] - 1] != nums[i]:tmp = nums[i]# 注意要保持这个位置loc = nums[i] - 1nums[i] = nums[nums[i] - 1]nums[loc] = tmpfor idx, val in enumerate(nums, 1):if val != idx:res.append(val)return res
思路二:
class Solution:def findDuplicates(self, nums: List[int]) -> List[int]:res = []for i in range(len(nums)):loc = abs(nums[i]) - 1if nums[loc] < 0:res.append(loc + 1)nums[loc] = -nums[loc]return res