leetcode [17.19] 消失的两个数字

---

Contact me:

Blog -> https://cugtyt.github.io/blog/index
Email -> cugtyt@qq.com
GitHub -> Cugtyt@GitHub

---

给定一个数组，包含从 1 到 N 所有的整数，但其中缺了两个数字。你能在 O(N) 时间内只用 O(1) 的空间找到它们吗？

以任意顺序返回这两个数字均可。

示例 1:

输入: [1]
输出: [2,3]

示例 2:

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

提示：

nums.length <= 30000

思路：添加两个值，然后把每个值按照索引排序，发现有值不存在表示缺失。

class Solution:
    def missingTwo(self, nums: List[int]) -> List[int]:
        for i in range(len(nums)):
            nums[i] -= 1
        nums.extend([-1, -1])
        for i in range(len(nums)):
            while i != nums[i] and nums[i] != -1:
                j = nums[i]
                nums[i], nums[j] = nums[j], nums[i]
        ans = []
        for i in range(len(nums)):
            if nums[i] == -1:
                ans.append(i + 1)
        if len(ans) == 0:
            ans = [len(nums) - 1, len(nums)]
        if len(ans) == 1:
            ans.append(len(nums))
        return ans