Leetcode 268. 缺失数字（Missing Number）

Leetcode 268. 缺失数字

1 题目描述（Leetcode题目链接）

  给定一个包含 $0,1,2,\cdots ,n$ 中 n 个数的序列，找出 $0\cdots n$ 中没有出现在序列中的那个数。

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

输入: [9,6,4,2,3,5,7,0,1]
输出: 8

说明：你的算法应具有线性时间复杂度。你能否仅使用额外常数空间来实现?

2 题解

  转述一下题目就是，从$0$到$n$这$n+1$个数中抽出去一个组成数组，问你抽出去的是哪一个。实现的方法多种多样，但是题目对时间和空间做了限制，所以不能使用排序或哈希表。这里给出几种做法。
  首先是数学方法，我们都知道等差数列的求和公式，那么$1$到$n$个数的和为$\frac{(1+n)n}{2}$，这个求和减去数组中元素的和就是缺失的那个数字。

class Solution:
    def missingNumber(self, nums: List[int]) -> int:
        n = len(nums)
        return (1 + n)*n // 2 - sum(nums)

  第二个方法其实也是数学方法，是用异或的方法，两个相同的数异或得0，这样，从数组第一个元素开始每次异或数组元素和其下标，最后得到的数就是缺失的数字。

class Solution:
    def missingNumber(self, nums: List[int]) -> int:
        retv = len(nums)
        for i in range(len(nums)):
            retv ^= i
            retv ^= nums[i]
        return retv

  第三个方法类似于哈希表，假设我们新开了一个数组长度为$n+1$，我们便可以在遍历数组的同时填充新数组，最后新数组中没有被填充的位置的下标就是缺失的数字。那么现在我们不新开数组，将原数组加个新元素$-1$，从头开始遍历数组，每遇到一个元素，我们就将其放到下标等于它的地方，最后这个$-1$所在的下标就是缺失的数字。

class Solution:
    def missingNumber(self, nums: List[int]) -> int:
        nums += [-1]
        retv = 0
        for i in range(len(nums)):
            while nums[i] != i:
                nums[nums[i]], nums[i] = nums[i], nums[nums[i]]
                if nums[i] == -1:
                    retv = i
                    break
        return retv