[leetcode]Missing Number

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

Example 1:

Input: [3,0,1]
Output: 2

分析:

找出数组中0~n-1缺失的数字。可以对数组排序,如果排好序的两个数字之差不为1,则中间的数就是缺失的数字;如果遍历到最后两两之间都差1,则缺的是最大的数字;如果数组首位不为0,则缺失的数字为0。

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

当然还有更简单的做法,即把数组里的所有元素相加,与0~n-1之间的整数之和相比,两者之差即为少的数。

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

猜你喜欢

转载自blog.csdn.net/weixin_41814716/article/details/83987539