LeetCode-Python-645. 错误的集合

集合 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]

LeetCode-Python-645. 错误的集合

猜你喜欢