LeetCode-Python-645. 错误的集合

集合 S 包含从1到 n 的整数。不幸的是,因为数据错误,导致集合里面某一个元素复制了成了集合里面的另外一个元素的值,导致集合丢失了一个整数并且有一个元素重复。

给定一个数组 nums 代表了集合 S 发生错误后的结果。你的任务是首先寻找到重复出现的整数,再找到丢失的整数,将它们以数组的形式返回。

示例 1:

输入: nums = [1,2,2,4]
输出: [2,3]

注意:

  1. 给定数组的长度范围是 [2, 10000]。
  2. 给定的数组是无序的。

第一种思路:

桶排序,新开一个数组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]

 

上一篇:556. Next Greater Element III


下一篇:HL-D793G七位蓝色+PT6311驱动