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;
}
};