集合 S
包含从1到 n
的整数。不幸的是,因为数据错误,导致集合里面某一个元素复制了成了集合里面的另外一个元素的值,导致集合丢失了一个整数并且有一个元素重复。
给定一个数组 nums
代表了集合 S
发生错误后的结果。你的任务是首先寻找到重复出现的整数,再找到丢失的整数,将它们以数组的形式返回。
示例 1:
输入: nums = [1,2,2,4] 输出: [2,3]
注意:
- 给定数组的长度范围是 [2, 10000]。
- 给定的数组是无序的。
第一种思路:
桶排序,新开一个数组n,先全部置0,
然后线性扫描nums, 对于nums里的元素digit, n[digit] = digit,如果赋值之前n[digit]不为空,就说明digit重复了。
扫描完成一趟后,再扫一次n,还是0的那个元素的下标就是没出现过的。
class Solution(object):
def findErrorNums(self, nums):
"""
:type nums: List[int]
:rtype: List[int]
"""
n = [0 for i in range(1, len(nums) + 1)]
res = list()
for i, digit in enumerate(nums):
if n[digit - 1] == 0:
n[digit - 1] = digit
else:
res.append(digit)
for i, x in enumerate(n):
if x == 0:
res.append(i + 1)
break
return res
第二种思路:
类似第一种思路,略有变化,
n[digit] 每次 + 1,所以最后是2的元素下标就是重复了的,是0的元素下标就是没出现过的。
class Solution(object):
def findErrorNums(self, nums):
"""
:type nums: List[int]
:rtype: List[int]
"""
n = [0 for i in range(1, len(nums) + 1)]
for i, digit in enumerate(nums):
n[digit - 1] += 1
for i, x in enumerate(n):
if x == 2:
twice = i + 1
elif x == 0:
never = i + 1
return [twice, never]