找到所有数组中消失的数字
题目
给定一个范围在 1 ≤ a[i] ≤ n ( n = 数组大小 ) 的 整型数组,数组中的元素一些出现了两次,另一些只出现一次。
找到所有在 [1, n] 范围之间没有出现在数组中的数字。
您能在不使用额外空间且时间复杂度为O(n)的情况下完成这个任务吗? 你可以假定返回的数组不算在额外空间内。
示例:
输入:
[4,3,2,7,8,2,3,1]
输出:
[5,6]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-all-numbers-disappeared-in-an-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解决思路
不符合题意——使用哈希表
把数组中的数字映射到哈希表中,利用for循环遍历限定范围内的数字是否出现在哈希表中,将不存在的数字添加到返回数组中。
学习思路
原地修改
关键点
- n 是 数组长度,且数组元素范围是[1,n],且数组元素存在重复
- 另一方面,数组索引范围是[0,n - 1]
- 不使用额外空间,即只能在数组内部操作,思考如下:
- 每个数组元素值 - 1 对应着唯一的数组索引;
- 例如,当数组中存在 2 ,则数组中 对应 1 位置上的数必然存在;
- 于是用某种标记过滤这些位置上的值,没有被标记意味着对应的索引值 + 1 的数字在数组中不存在;
- 标记采用取数组元素的负值,原数组元素均为正整数,这样数组元素的绝对值不变,能通过绝对值的方式找到正确的数组元素值;
- 由于存在重复元素,所以标记时加判断,若为大于零的数字说明没有重复,这时才加标记——变为负数,已变成负数则不做改变。
代码
使用哈希表(非适用)
class Solution:
def findDisappearedNumbers(self, nums: List[int]) -> List[int]:
Hash = {}
s = []
for i in nums:
Hash[i] = 1
for i in range(1,len(nums) + 1):
if i not in Hash:
s.append(i)
return s
-
结果
-
时间复杂度O(N),遍历数组
-
空间复杂度O(N),哈希表
原地修改
class Solution:
def findDisappearedNumbers(self, nums: List[int]) -> List[int]:
s = []
for i in nums:
i = abs(i)
if nums[i - 1] > 0:
nums[i - 1] *= -1
for i in range(len(nums)):
if nums[i] > 0:
s.append(i + 1)
return s
-
结果
-
时间复杂度O(N),遍历数组
-
空间复杂度O(1)
问题解决
-
IndexError:list assignment index out of range
list 为空列表时,直接扩充元素会报错,使用list.append(obj)添加; -
list 也能起到和 哈希表一样的作用,但是后者查找的时间复杂度为O(1),前者查找语句另加for循环时间复杂度为O(N^2);
-
原地修改时,第一次遍历就标记原数组中的值,会存在原数组后遍历的值已被修改,破坏了初始完整的数组,由于变成负值,标记的位置也不一样,背离解题初衷,所以在遍历时为保证正确索引,利用数组元素的绝对值 - 1得到。
学习总结
- 新的思路:数组索引和数组元素的联系技巧,原地修改,对空间复杂度友好,在特定数组范围和特殊数组长度情况下,数组索引和数组元素存在线性对应关系;
- 对环结构的理解有帮助。