leetcode_442 数组中重复的数据

181 阅读1分钟

要求

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

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

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

示例:

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

输出:
[2,3]

详解代码

class Solution:
    def findDuplicates(self, nums: List[int]) -> List[int]:
        if not nums:
            return []
        result = []
        for _,num in enumerate(nums):
            index = abs(num) - 1
            if nums[index] < 0:
                result.append(index + 1)
            nums[index] *= -1
        return result

image.png

解题思路:这个有点像翻牌子的游戏,当我们从前向后搜索的时候,我们遇到没有见到过的数字,他的换算后的索引的牌子,都是不相同的,我们将其反转过去,就是乘-1的操作,当我们遇到相同的数字的时候,我们换算后的索引都是相同的,所以我们观察是否被翻过牌子就知道,这个数字是不是出现过,一次循环即可搞定所有的事,时间复杂度控制在O(n)。

当前解法的缺陷是:当给定数据中的数字大小超过列表的长度的时候,我们的程序可能出现out of range的情况,后续改进