[Leetcode][python]First Missing Positive/缺失的第一个正数

题目大意

给定一个未经排序的数组，寻找第一个缺失的正整数
你的算法应该满足O(n)时间和常数空间复杂度

解题思路

参考：http://bookshadow.com/weblog/2015/09/14/leetcode-first-missing-positive/

尽可能地把数组中不大于n（n为数组长度）的正整数放置到下标+1与其数值相同的位置上

第一个下标+1与数值不同的数字，即为所求。

例如数组nums = [3,4,-1,1]，调整位置后的结果为：[1,-1,3,4]

除第二个数字外，其余数字均满足nums[i] = i + 1，因此返回2

代码

class Solution(object):
    def firstMissingPositive(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        n = len(nums)
        for i in range(n):
            # print nums[i], i + 1, nums[nums[i] - 1]
            while nums[i] > 0 and nums[i] <= n and nums[i] != i + 1 and nums[i] != nums[nums[i] - 1]:
                nums[nums[i] - 1], nums[i] = nums[i], nums[nums[i] - 1]
        # print nums
        for i in range(n):
            if i + 1 != nums[i]:
                return i + 1
        return n + 1

总结

tricky题目