目录
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每日一题总目录(动态更新。。。)