leetcode 41. First Missing Positive（思维)

First Missing Positive
Hard

2902

741

Add to List

Share
Given an unsorted integer array, find the smallest missing positive integer.

Example 1:

Input: [1,2,0]
Output: 3
Example 2:

Input: [3,4,-1,1]
Output: 2
Example 3:

Input: [7,8,9,11,12]
Output: 1
Note:

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

这个题目其实是287. Find the Duplicate Number的一个变形，为了达到O(N)时间复杂度，O(1)空间复杂度，我们必须要用原数组来进行一个索引数组。对于这个题目，我们可以肯定，结果是在[1,nums.size()+1]这个范围的。而且结果为nums.size()+1的只有一种情况，所以我们可以不考虑负数和大于nums.size()的数，然后通过交换把其放到属于自己的位置，即nums[i]-1;的位置。最后一次判断当前位置是否符合nums[i] == i+1;
代码如下：

class Solution {
public:
    int firstMissingPositive(vector<int>& nums) {
        for(int i=0;i<nums.size();++i){
            while(nums[i] != i+1){
                if(nums[i] <= 0 || nums[i] > nums.size() || nums[i] == nums[nums[i]-1]){
                //这里第三个条件不能少，用来判断存在相同的数，并且一个已经放在了正确的位置。
                    break;
                }
                swap(nums[i],nums[nums[i]-1]);
            }
        }
        for(int i=0;i<nums.size();++i){
            if(nums[i] != i+1){
                return i+1;
            }
        }
        return nums.size()+1;
    }
};