[LeetCode]71. Missing Number缺失的数

Given an array containing n distinct numbers taken from

0, 1, 2, ..., n

, find the one that is missing from the array.

For example,
Given nums =

[0, 1, 3]

return

2

.

Note:
Your algorithm should run in linear runtime complexity. Could you implement it using only constant extra space complexity?

Credits:
Special thanks to @jianchao.li.fighter for adding this problem and creating all test cases.

Subscribe to see which companies asked this question

解法1：不考虑常数空间复杂度的限制，一个简单的解法是初始化一个n+1个元素的vector<int> v(n+1,-1)，然后遍历输入数组，将v对应下标处的值设为输入数组当前值。最后对v遍历一遍看哪个下标处的值还是-1即可。时间空间复杂度都是O(n)。可以用bitset来压缩空间复杂度。

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

解法2：先将输入数组排序，然后扫描一遍找到第一个不在位置的元素即可。时间空间复杂度取决于所用排序算法。

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

解法3：因为这n个数是从[0,n]中抽取的，因此分别求和：sum1=(0+n)*(n+1)/2为所有n-1个数的和，遍历输入数组可以求得抽取的n个数的和sum2，两者相减即得缺失的那个数。时间复杂度O(n)，空间复杂度O(1)。

class Solution {
public:
int missingNumber(vector<int>& nums) {
int n = nums.size();
int sum1 = n * (n + 1) / 2;
int sum2 = accumulate(nums.begin(), nums.end(), 0);
return sum1 - sum2;
}
};

解法4：结合题目Single Number可以想到基于位操作的解法：将输入数组与[0,n]异或，最后得到的值即是缺失的那个。因为输入数组元素为n个，任何数异或0都是它本身，因此我们只需从1开始到n结束即可。时间复杂度O(n)，空间复杂度O(1)。

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