Array&Matrix-6.错误的集合(645)

1.题目

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

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

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

注意:
给定数组的长度范围是 [2, 10000]。
给定的数组是无序的。

2. 思路

(1)先将无序数组排序，找到重复的数， nums[i] == nums[i+1]；
然后在[1,n]里，找缺失的数。i+1 not in nums

class Solution:
    def findErrorNums(self, nums):
        nums.sort()
        r = 0
        s = 0
        for i in range(len(nums)-1):
            if nums[i] == nums[i+1]:
                r = nums[i]
                break;
        for num in range(len(nums)):
            if num+1 not in nums:
                s = num + 1
                break;
        return [r,s]

(2) 将两个循环合并
需要考虑特殊情况，[1,1] 判断数组最后一位数是否等于长度

class Solution:
    def findErrorNums(self, nums):
        nums.sort()
        n = len(nums)
        rep = 0
        miss = 1
        for i in range(1, n):
            if nums[i] == nums[i-1]:
                rep = nums[i]
            elif nums[i] > nums[i-1] + 1:
                miss = nums[i-1] + 1
        return [rep, n if nums[n-1] != n else miss]