题目描述
Given an unsorted integer array nums, return the smallest missing positive integer.
You must implement an algorithm that runs in O(n) time and uses constant extra space.
解题思路
首先我觉得这道题目非常值得一做且值得思考。O(n) 的时间复杂度我首先想到的是用字典,存储所有的正数然后利用hash的常数查找时间的优势控制时间复杂度。
很遗憾,上述方法空间开销太大,不符合题意。但是字典的思想还是会用到,我们能不能利用原始数组去构造一个比较像 字典的东西?设原始数组长度为 n, 我们最 care 的数组内容是 [1, n] 内的数字。有了这个前提,我们可以构造一个类似字典的东西:key 为数组下标,value 为数组内值为 key + 1的数字。(注:当 value 找不到时,其值没有意义可忽略。)然后,我们根据上述规则,将数组中每个属于 [1,n] 的内容放入对应的键中,这道题基本就解出来了。
代码
class Solution:
def firstMissingPositive(self, nums: List[int]) -> int:
for i in range(len(nums)):
while 1 <= nums[i] and nums[i] < len(nums) and nums[i] != nums[nums[i] - 1]: # 让 nums[i] 放入 i + 1
temp = nums[nums[i] - 1]
nums[nums[i] - 1] = nums[i]
nums[i] = temp
for j in range(len(nums)):
if nums[j] != j + 1:
return j + 1
return len(nums) + 1