[LintCode] Find the Missing Number

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;
}
posted @ 2017-09-12 21:00  mioopoi  阅读(158)  评论(0编辑  收藏  举报