目录

0442:数组中重复的数据(★)

力扣第 442 题

题目

给你一个长度为 n 的整数数组 nums ,其中 nums 的所有整数都在范围 [1, n] 内,且每个整数出现 一次两次 。请你找出所有出现 两次 的整数,并以数组形式返回。

你必须设计并实现一个时间复杂度为 O(n) 且仅使用常量额外空间的算法解决此问题。

示例 1:

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

示例 2:

输入:nums = [1,1,2]
输出:[1]

示例 3:

输入:nums = [1]
输出:[]

提示:

  • n == nums.length
  • 1 <= n <= 105
  • 1 <= nums[i] <= n
  • nums 中的每个元素出现 一次两次

分析

#1

可以类似 0041,将正数都放在对应的位置上,出现两次的没法全放,即可返回。

1
2
3
4
5
6
7
class Solution:
    def findDuplicates(self, nums: List[int]) -> List[int]:
        for i,x in enumerate(nums):
            while x!=nums[x-1]:
                nums[i], nums[x-1] = nums[x-1], nums[i]
                x = nums[i]
        return [x for i,x in enumerate(nums) if x!=i+1]

89 ms

#2

还有个修改原数组的方法:

  • 遍历到数 x 时,将位置 x-1 取为负数
  • 如果位置 x-1 已经是负数,即说明 x 是第二次遇见了

解答

1
2
3
4
5
6
7
8
9
class Solution:
    def findDuplicates(self, nums: List[int]) -> List[int]:
        res = []
        for x in nums:
            if nums[abs(x)-1] > 0:
                nums[abs(x)-1] *= -1
            else:
                res.append(abs(x))     
        return res

71 ms