LeetCode #268 - Missing Number - Medium

Problem

Given an array containing n distinct numbers taken from 0, 1, 2, ..., n, find the one that is missing from the array.

Note:

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

Example

Given nums = [0, 1, 3] return 2.

---

Algorithm

整理一下题意：给定一个包括n个不同的数的数组，这个数组所有元素来自0，1，2，…，n，要求找到这个数组在N(0,n)缺少的数字。

首先是想到简单的遍历的方法。即给每个数组中每个元素标记为true，然后遍历0~n，若找到标记不是true的数，则这个数便是缺少的数字。

代码如下。

//版本1，遍历
class Solution {
public:
    int missingNumber(vector<int>& nums) {
        int n=nums.size();
        int miss;
        vector<bool> in(n+1,false);
        for(int i=0;i<n;i++){
            in[nums[i]]=true;
        }
        for(int i=0;i<=n;i++){
            if(in[i]==false){
                miss=i;
                break;
            }
        }
        return miss;
    }
};

然而题目提问能否在O(n)时间内使用常数空间完成，上面的做法因为开了一个长度为n的数组，而n是变量，所以所占用的空间不是常数空间。于是重新思考。

如果只能占用常数空间，说明不能开一个随n变化的数组，那么需要利用一些数学关系来找出缺少的数字。因为已知所有数字只能在0~n中取得，有且仅有一个数字缺失，且该数字属于0~n。于是可以尝试把0~n加起来的和与给定数组的和比较，二者差值即为缺失的数。

代码如下。

版本2，差值
class Solution {
public:
    int missingNumber(vector<int>& nums) {
        int sum_num=0;
        int sum_n=0;
        for(int i=0;i<nums.size();i++){ sum_num+=nums[i]; }
        for(int i=0;i<nums.size()+1;i++){sum_n+=i; }
        return sum_n-sum_num;
    }
};

然后继续思考，能不能找到更好解法呢？ 后来在DISCUSS里面浏览时，发现一种利用异或的解法，也很巧妙。其原理是如果两个数相等，则异或结果为0,；否则异或结果为1。利用这个性质来找出缺少的数字。

代码如下。

//版本3，异或
class Solution {
public:
    int missingNumber(vector<int>& nums) {
        int result=0;
        for(int i=0;i<nums.size();i++){
            result^=((i+1)^nums[i]);
        }
        return result;
    }
};