41. 缺失的第一个正数

· 约 2 分钟阅读 · 次阅读 leetcode

给你一个未排序的整数数组 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


直观的思路:哈希表

但是时间空间复杂度都是O(n),空间复杂度不满足题目的要求

class Solution:
    def firstMissingPositive(self, nums: List[int]) -> int:
        exists = collections.defaultdict(int)
        for i in range(len(nums)):
            exists[nums[i]] = 1
        
        for i in range(len(nums)+1):
            if exists[i+1] == 1:
                continue
            else:
                return i+1
        return len(nums)+1

改造当前数组为哈希表

数组长度为n,则数组中的数,如果存储的是[1,n], 则第一个缺失的数是n+1 否则缺失的数一定在[1,n]中的某一个。

class Solution:
    def firstMissingPositive(self, nums: List[int]) -> int:
        n = len(nums)
        for i in range(n):
            # 把非正数都改为一个特殊的数,注意0也要修改
            if nums[i] <= 0:
                nums[i] = n + 1
        
        
        for i in range(n):
            # 数有可能已经被前序循环修改为负数了
            num = abs(nums[i])
            if num <= n:
                # print(num, nums)
                # 可能有重复的数
                nums[num-1] = -abs(nums[num-1])
        
        for i in range(n):
            if nums[i] >= 0:
                return i+1
        
        return n+1

交换方式

class Solution:
    def firstMissingPositive(self, nums: List[int]) -> int:
        def swap(nums, i, j):
            nums[i], nums[j] = nums[j], nums[i]

        n = len(nums)
        for i in range(n):
            # 多次交换, 注意死循环
            while 0 <  nums[i] <= n:
                index = nums[i] - 1
                if nums[index] == nums[i]:
                    break
                else:
                    swap(nums, i, index)

        
        for i in range(n):
            if nums[i] != i+1:
                return i+1
        
        return n+1