leetcode 41. 缺失的第一个正数

Posted 2021-03-03 wz-archer

给你一个未排序的整数数组，请你找出其中没有出现的最小的正整数。

示例 1:

输入: [1,2,0]
输出: 3
示例 2:

输入: [3,4,-1,1]
输出: 2
示例 3:

输入: [7,8,9,11,12]
输出: 1
 

提示：

你的算法的时间复杂度应为O(n)，并且只能使用常数级别的额外空间。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/first-missing-positive
解：首先借用计数排序的思想，可以在声请n个空间记录每个数字出现的次数，然后从1开始遍历，第一个次数为0的数字字就是我们需要的数字。这样做空间复杂度为n，我们又不能用原来的数组表示每个数字出现的次数，但是在本题中，我只需要知道每个数字是否出现过就好了，所以当nums[2]=6时，我们让nums[2]和nums[6]交换位置，当nums[i]==i时，表示这个数字出现过。

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