[leetcode] 41. First Missing Positive 解题报告

题目链接: https://leetcode.com/problems/first-missing-positive/

Given an unsorted integer array, find the first missing positive integer.

For example,
Given [1,2,0] return 3,
and [3,4,-1,1] return 2.

Your algorithm should run in O(n) time and uses constant space

思路: 遍历一遍数组, 将元素交换到其索引位置, 然后再遍历一遍数组, 如果当前位置的索引和其值不一样, 则返回这个值.

另外还有一些特殊情况, 就是最后nums[0]的位置, 这个位置可能会隐藏最大的值, 因此最后需要判断一下这个位置, 如果这个位置的值和我们找到的最大值相同则返回+1的值, 当然如果为数组再增加一个容量会更简单, 这样所有的值都能够交换到其应该在的地方.

代码如下:

class Solution {
public:
    int firstMissingPositive(vector<int>& nums) {
        if(nums.size() ==0) return 1;
        int len = nums.size(), ans = 0;
        for(int i = 0; i < len; i++)
        {
            while(nums[i]!=i && nums[i]<len && nums[i]>=0 && nums[nums[i]]!=nums[i]) 
                swap(nums[i], nums[nums[i]]);
        }
        for(ans = 1; ans < len; ans++) if(ans!=nums[ans]) break;
        return nums[0]==ans?ans+1:ans;
    }
};