Leetcode0442. 数组中重复的数据(medium)

目录

1. 问题描述

2. 思路与算法

3. 代码实现

---

1. 问题描述

给你一个长度为 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 <= 10^5
• 1 <= nums[i] <= n
• nums 中的每个元素出现 一次 或 两次

2. 思路与算法

        时间复杂度为 意味着对输入数组一遍扫描过去针对每个元素只能做常数量级操作。 

        最直观的做法是扫描数组，对于，查询它是否存在于到，但是这样要求的复杂度为，显然不可行。

        先进行排序再进行查询，查询动作本身可以实现，但是排序需要，所以也不满足题目要求。

        用哈希表存储每个出现过的数，然后遍历数组，如果发现有已经出现在哈希表中的数即加入答案列表，这样需要空间复杂度为，也不满足题目要求。要满足常数空间复杂度要求，需要找到办法直接利用原始输入数组的空间来存储中间查询信息。

        由于输入数组中的元素值的范围为，考虑输入数组的索引用来存储在原数组中出现过次数（或者说是否出现过）的信息，从空间容量来说是足够的。但是如何做到对在查询过程中原地修改而又不丢失还没有查询到的数的信息呢？

        可以考虑给nums中数添加负号来实现以上目的。

        以示例1的数据[4,3,2,7,8,2,3,1]进行说明，第1个数为4，对应的索引为3，将索引3处的数改为原来的相反数，即。这样既保留了原来存储在索引3处的数，又记录了出现过的信息。

        如果下次再碰到4的话，一看nums[3]为负数，就表明它已经出现过，可以加入答案列表。

        对数组[4,3,2,7,8,2,3,1]进行一次遍历后，数组内容的变化情况如下所示。

        k = 0 : [4, 3, 2, -7, 8, 2, 3, 1]

        k = 1 : [4, 3, -2, -7, 8, 2, 3, 1]

        k = 2 : [4, -3, -2, -7, 8, 2, 3, 1]

        k = 3 : [4, -3, -2, -7, 8, 2, -3, 1]

        k = 4 : [4, -3, -2, -7, 8, 2, -3, -1]

        k = 5 : [4, -3, -2, -7, 8, 2, -3, -1]

        k = 6 : [4, -3, -2, -7, 8, 2, -3, -1]

        k = 7 : [-4, -3, -2, -7, 8, 2, -3, -1]

        需要注意的是，由于在遍历过程中数组内的元素会被符号取反，所以始终要对数组值取绝对值来获取原始数组中对应位置的数。

3. 代码实现

class Solution:
    def findDuplicates(self, nums: List[int]) -> List[int]:
        ans = []
        for k in range(len(nums)):
            if nums[abs(nums[k])-1] < 0:
                ans.append(abs(nums[k]))
            else:
                nums[abs(nums[k])-1] = - nums[abs(nums[k])-1]
            #print('k = ',k,': ',nums)
        return ans

        执行用时：104 ms, 在所有 Python3 提交中击败了37.18%的用户

        内存消耗：21.4 MB, 在所有 Python3 提交中击败了33.37%的用户

        回到总目录：笨牛慢耕的Leetcode每日一题总目录(动态更新。。。)