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

leetcode 645. 错误的集合
看到这个时间,也是醉了。

解法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]

leetcode 645. 错误的集合