【leet-code清晰解题思路💯✅】41. 缺失的第一个正数

26 阅读1分钟

题目描述

给你一个未排序的整数数组 nums ,请你找出其中没有出现的最小的正整数。 请你实现时间复杂度为 O(n) 并且只使用常数级别额外空间的解决方案。

示例 1:

输入: nums = [1,2,0]
输出: 3

示例 2:

输入: nums = [3,4,-1,1]
输出: 2

示例 3:

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

提示:

  • 1 <= nums.length <= 5 * 105
  • -231 <= nums[i] <= 231 - 1

解题思路

  • 最直接的思路是开一个新数组,计数排序。但是题目要求我们只能用常数级空间。所以我们需要原地算法,原地计数排序,从前向后遍历:
    • 如果当前值不等于下标,且在范围内,就把他放到正确的位置。不断循环,所有数都放回了正确的位置。还要注意防止死循环。
    • 然后再从前向后遍历,如果值与下标不符说明不存在,返回答案即可。
func firstMissingPositive(nums []int) int {
   for i := 0; i < len(nums); i++ {
      for nums[i] > 0 && nums[i] < len(nums) && nums[i] != i+1 && nums[i] != nums[nums[i]-1] {
         nums[i], nums[nums[i]-1] = nums[nums[i]-1], nums[i]
      }
   }
   for i := 0; i < len(nums); i++ {
      if nums[i] != i+1 {
         return i + 1
      }
   }
   return len(nums) + 1
}

image.png