Leetcode 41.缺失的第一个正数题解

Leetcode 41.缺失的第一个正数题解

题目链接：

https://leetcode-cn.com/problems/first-missing-positive/

题目描述

给你一个未排序的整数数组，请你找出其中没有出现的最小的正整数。
提示：
你的算法的时间复杂度应为O(n)，并且只能使用常数级别的额外空间。

思路

因为这道题要求时间复杂度是O(n)，所以可以先思考一下遍历一遍数组我们可以得到什么，显然我们可以得到数组中大于0的最大值和最小值。于是我就想到生成一个hash来记录数组中出现的正数，然后再从零找没有出现的最小正整数。但是空间复杂度要求O(1)，就说明肯定不能再生成一个空间存储hash表，但不用hash又达不到O(n)的时间复杂度。仔细思考一下，发现最小正整数肯定不会大于数组的长度，因此不妨把初始的数组改为hash表，如果数组中的数字小于数组长度，那么就把他和对应index上数字交换（除非相同），反之置为-1。这样遍历一遍就可以保证每个index上面的数字都与之相同或者为-1，然后最小的正整数就显而易见了。

代码

class Solution(object):
    def firstMissingPositive(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        i = 0
        while i < len(nums):
            if nums[i] < 1:
                i += 1
            elif nums[i] > len(nums):
                nums[i] = -1
                i += 1
            else:
                if nums[i] == i + 1:
                    i += 1
                else:
                    tmp = nums[i]
                    if tmp == nums[tmp - 1]:
                        nums[i] = -1
                        i += 1
                    else:
                        nums[i], nums[tmp - 1] = nums[tmp - 1], nums[i]
        for i, n in enumerate(nums):
            if n < 1:
                return i + 1
        return len(nums) + 1