leetcode #41. 缺失的第一个正数

题目描述:

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

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

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

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

说明:

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

解题思路:

首先,因为所求的为最小正数,则对于数组中的负数,不会影响题目的结果,故可以把负数直接置0即可。
其次,因为最小的正数是从1开始,则得到的结果不超过数组的长度,那么对于数组中所有大于数组长度的值,不会影响得到的结果,故同样直接置0.
经过上述操作后,数组中的元素均大于等于0,且小于等于数组长度,则从头遍历数组,若该值的绝对值大于0且小于等于数组长度,修改数组下标为该绝对值-1的值,有3种情况:
(1)若该值为0,修改该值为一个负数(这个负数只要绝对值大于数组长度即可)。
(2)若该值为负数,说明已被修改过,此时跳过即可。
(3)若该值为正数,置为它的相反数。
最后,对数组从头开始遍历,第一个大于等于0的值对应的下标加一即为所求结果。

代码:

class Solution:
    def firstMissingPositive(self, nums):
        n = len(nums)
        res = 1
        if n == 0:
            return res
        for num in range(0,n):
            if nums[num] < 0:
                nums[num] = 0
            if nums[num] > n:
                nums[num] = 0
        flag = -(n+1)

        for num in nums:
            if num != 0 and num != flag:
                if nums[abs(num)-1] == 0:
                    nums[abs(num)-1] = flag
                elif nums[abs(num)-1] > 0:
                    nums[abs(num)-1] *= -1

        while res <= n and nums[res-1] < 0:
            res += 1
        return res

你可能感兴趣的:(homework,homework)