面试题 17.04. 消失的数字
数组nums包含从0到n的所有整数,但其中缺了一个。请编写代码找出那个缺失的整数。你有办法在O(n)时间内完成吗?
题解:利用求和公式,算到0至(n+1)之和,减去所有数字之和。
class Solution {
public:
int missingNumber(vector<int>& nums) {
int n = nums.size();
int max = n * (n + 1) >> 1; //右移1和除以2是相同效果
int sum = 0;
for (int i=0; i<n; ++i) {
sum += nums[i];
}
return max - sum;
}
};
/*
利用C++标准库 #include <numeric>
template <class InputIterator, class T>
T accumulate (InputIterator first, InputIterator last, T init)
init:Initial value for the accumulator.
*/
class Solution {
public:
int missingNumber(vector<int>& nums) {
int n = nums.size();
int max = n * (n + 1) >> 1;
int sum = accumulate(nums.begin(), nums.end(), 0);
return max - sum;
}
};