leetcode 645. 错误的集合
题目描述
集合 S 包含从1到 n 的整数。不幸的是,因为数据错误,导致集合里面某一个元素复制了成了集合里面的另外一个元素的值,导致集合丢失了一个整数并且有一个元素重复。
给定一个数组 nums 代表了集合 S 发生错误后的结果。你的任务是首先寻找到重复出现的整数,再找到丢失的整数,将它们以数组的形式返回。
示例 1:
输入: nums = [1,2,2,4]
输出: [2,3]
注意:
给定数组的长度范围是 [2, 10000]。
给定的数组是无序的。
解法一思路
两个任务,第一去重,第二找到正常排序丢失的元素,遍历给的nums数组,若出现在自然数组中,就去掉自然数组中的元素,剩下的是丢失的元素,在添加一个临时数组,存放遍历的元素, 找出重复的元素,O(n*1) 的复杂度。
解法1
class Solution:
def findErrorNums1(self, nums):
tdict = []
res = []
origin = {}
for k in range(1,len(nums)+1):
origin[k] = 0
for i in nums:
if i in tdict:
res.append(i)
else:
tdict.append(i)
if i in origin:
del origin[i]
for j in origin:
res.append(j)
return res
看到这个时间,也是醉了。
解法2
- 思路:
计算 自然数、给定nums 、去重之后 三个数组的sum ,两两相减,得到重复和丢失数 时间复杂度 O(1)
def findErrorNums(self, nums):
# 自然数 S
# nums N
# set L
S = len(nums)* (len(nums)+1) // 2
N = sum(nums)
L = sum(set(nums))
return [N-L,S-L]