力扣刷题学习268. 丢失的数字（C++）

题目描述

给定一个包含 [0, n] 中 n 个数的数组 nums ，找出 [0, n] 这个范围内没有出现在数组中的那个数。

测试用例

示例一：

输入：nums = [3,0,1] 输出：2 解释：n = 3，因为有 3 个数字，所以所有的数字都在范围 [0,3] 内。2
是丢失的数字，因为它没有出现在 nums 中。

示例二：

输入：nums = [0,1]
输出：2
解释：n = 2，因为有 2 个数字，所以所有的数字都在范围 [0,2] 内。2 是丢失的数字，因为它没有出现在 nums 中。

示例三：

输入：nums = [9,6,4,2,3,5,7,0,1]
输出：8
解释：n = 9，因为有 9 个数字，所以所有的数字都在范围 [0,9] 内。8 是丢失的数字，因为它没有出现在 nums 中。

提示：

• n == nums.length
• 1 <= n <= 104
• 0 <= nums[i] <= n
• nums 中的所有数字都 独一无二

解答

首先看到这题，我们第一反应可能是直接暴力查询，但这样复杂度是log(n^2)，
我们可以稍微优化一下，先排序，再查询。
方案一：排序

class Solution {
public:
    int missingNumber(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        int n = nums.size();
        for (int i = 0; i < n; i++) {
            if (nums[i] != i) {
                return i;
            }
        }
        return n;
    }
};

同时还可以用空间换时间，采用哈希集合。
方案二：哈希

class Solution {
public:
    int missingNumber(vector<int>& nums) {
        unordered_set<int> set;
        int n = nums.size();
        for (int i = 0; i < n; i++) {
            set.insert(nums[i]);
        }
        int missing = -1;
        for (int i = 0; i <= n; i++) {
            if (!set.count(i)) {
                missing = i;
                break;
            }
        }
        return missing;
    }
};

复杂度分析

方案一：

• 时间复杂度：O(n \log n)O(nlogn)，其中 nn 是数组 \textit{nums}nums 的长度。排序的时间复杂度是O(n \log n)O(nlogn)，遍历数组寻找丢失的数字的时间复杂度是 O(n)O(n)，因此总时间复杂度是 O(n \logn)O(nlogn)。
• 空间复杂度：O(\log n)O(logn)，其中 nn 是数组 \textit{nums}nums的长度。空间复杂度主要取决于排序的递归调用栈空间。

方案二：

• 时间复杂度：O(n)O(n)，其中 nn 是数组 \textit{nums}nums 的长度。遍历数组 \textit{nums}nums 将元素加入哈希集合的时间复杂度是 O(n)O(n)，遍历从 00 到 nn 的每个整数并判断是否在哈希集合中的时间复杂度也是 O(n)O(n)。
• 空间复杂度：O(n)O(n)，其中 nn 是数组 \textit{nums}nums 的长度。哈希集合中需要存储 nn 个整数。