http://www.lintcode.com/zh-cn/problem/find-the-missing-number/#
先做一轮桶排序,把每个元素放在它应该在的位置上
(只要nums[i] != i
并且nums[i] < n
,就执行交换:swap(nums[i], nums[nums[i]])
,注意nums[i] < n
的条件不能漏掉,否则交换会导致数组越界访问);然后再扫描一次,看nums[i]
是否等于i
,不是就返回i
。
int missingNumber(vector<int>& nums) {
if (nums.empty()) return 0;
int n = nums.size();
for (int i = 0; i < n; ++i) {
while (nums[i] != i && nums[i] < n) {
std::swap(nums[i], nums[nums[i]]);
}
}
for (int i = 0; i < n; ++i) {
if (nums[i] != i) {
return i;
}
}
return n;
}