【程序员面试金典】面试题 17.04. 消失的数字

题目

数组nums包含从0到n的所有整数，但其中缺了一个。请编写代码找出那个缺失的整数。你有办法在O(n)时间内完成吗？

注意：本题相对书上原题稍作改动

示例 1：

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

示例 2：

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

思路

新建一个辅助数组，原数组出现的数字对应辅助数组下标为1，其余为0。

代码

时间复杂度：O(n)
空间复杂度：O(n)

class Solution {
public:
    int missingNumber(vector<int>& nums) {
        int size = nums.size();
        vector<int> vec(size + 1, 0);
        for (auto a : nums) {
            vec[a] = 1;
        }
        for (int i = 0; i < vec.size(); ++i) {
            if (vec[i] == 0) return i;
        }
        return -1;
    }
};