【数组】力扣41:缺失的第一个正数
给你一个未排序的整数数组 nums ,请你找出其中没有出现的最小的正整数。
请你实现时间复杂度为 O(n) 并且只使用常数级别额外空间的解决方案。
示例1:
输入:nums = [1,2,0]
输出:3
示例2:
输入:nums = [7,8,9,11,12]
输出:1
分析:
如果本题没有额外的时空复杂度要求,那么就很容易实现:
方法1:将数组所有的数放入哈希表,随后从 1 开始依次枚举正整数,并判断其是否在哈希表中;
方法2:从 1 开始依次枚举正整数,并遍历数组,判断其是否在数组中。
如果数组的长度为 N,那么第一种做法的时间复杂度为 O(N),空间复杂度为 O(N);第二种做法的时间复杂度为 O(N^2),空间复杂度为 O(1)。但它们都不满足时间复杂度为 O(N) 且空间复杂度为 O(1)。
∴ 「真正」满足时间复杂度为 O(N) 且空间复杂度为 O(1) 的算法是不存在的,但是我们可以退而求其次:利用给定数组中的空间来存储一些状态。也就是说,如果题目给定的数组是不可修改的,那么就不存在满足时空复杂度要求的算法;但如果我们可以修改给定的数组,那么是存在满足要求的算法的。
方法一:哈希表
哈希表是一个可以支持快速查找的数据结构:给定一个元素,我们可以在 O(1) 的时间查找该元素是否在哈希表中。
对于一个长度为 N 的数组,其中没有出现的最小正整数只能在 [1,N+1] 中。这是因为如果 [1, N] 都出现了,那么答案是 N+1,否则答案是 [1, N] 中没有出现的最小正整数。这样一来,我们将所有在 [1, N] 范围内的数放入哈希表,也可以得到最终的答案。而给定的数组恰好长度为 N ,这让我们有了一种将数组设计成哈希表的思路:
对数组进行遍历,对于遍历到的数 x,如果它在 [1, N] 的范围内,那么就将数组中的第 x-1 个位置(注意:数组下标从 0 开始)打上「标记」。在遍历结束之后,如果所有的位置都被打上了标记,那么答案是 N+1,否则答案是最小的没有打上标记的位置加 1。
那么如何设计这个「标记」呢?由于数组中的数没有任何限制,因此这并不是一件容易的事情。但我们可以继续利用上面的提到的性质:由于我们只在意 [1, N] 中的数,因此我们可以先对数组进行遍历,把不在 [1, N] 范围内的数修改成任意一个大于 N 的数(例如 N+1)。这样一来,数组中的所有数就都是正数了,因此可以将「标记」表示为「负号」。算法的流程如下:
- 将数组中所有小于等于 0 的数修改为 N+1;
- 遍历数组中的每一个数 x,它可能已经被打了标记,因此原本对应的数为 |x|,其中 | | 为绝对值符号。如果 |x| ∈ [1, N],那么给数组中的第 |x| - 1 个位置的数添加一个负号。注意如果它已经有负号,不需要重复添加;
- 在遍历完成之后,如果数组中的每一个数都是负数,那么答案是 N+1,否则答案是第一个正数的位置加 1。
class Solution:
def firstMissingPositive(self, nums: List[int]) -> int:
n = len(nums)
for i in range(n):
if nums[i] <= 0:
nums[i] = n + 1
for i in range(n):
num = abs(nums[i])
if num <= n:
nums[num - 1] = -abs(nums[num - 1])
for i in range(n):
if nums[i] > 0:
return i + 1
return n + 1
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/first-missing-positive/solution/que-shi-de-di-yi-ge-zheng-shu-by-leetcode-solution/
时间复杂度:O(N),其中 N 是数组的长度。
空间复杂度:O(1)。
方法二:置换
除了打标记以外,我们还可以使用置换的方法,将给定的数组「恢复」成下面的形式:
如果数组中包含 x ∈ [1,N],那么恢复后,数组的第 x - 1 个元素为 x。
在恢复后,数组应当有 [1, 2, ..., N] 的形式,但其中有若干个位置上的数是错误的,每一个错误的位置就代表了一个缺失的正数。以题目中的示例二 [3, 4, -1, 1] 为例,恢复后的数组应当为 [1, -1, 3, 4],就可以知道缺失的数为 2。
那么如何将数组进行恢复呢?我们可以对数组进行一次遍历,对于遍历到的数 x = nums[i],如果 x ∈ [1,N],我们就知道 x 应当出现在数组中的 x - 1 的位置,因此交换 nums[i] 和 nums[x−1],这样 x 就出现在了正确的位置。在完成交换后,新的 nums[i] 可能还在 [1, N]的范围内,我们需要继续进行交换操作,直到 x ∈ [1,N]。
注意到上面的方法可能会陷入死循环。如果 nums[i] 恰好与 nums[x−1] 相等,那么就会无限交换下去。此时我们有 nums[i] = x = nums[x−1],说明 x 已经出现在了正确的位置。因此我们可以跳出循环,开始遍历下一个数。
由于每次的交换操作都会使得某一个数交换到正确的位置,因此交换的次数最多为 N,整个方法的时间复杂度为 O(N)。
class Solution:
def firstMissingPositive(self, nums: List[int]) -> int:
n = len(nums)
for i in range(n):
while 1 <= nums[i] <= n and nums[nums[i] - 1] != nums[i]:
nums[nums[i] - 1], nums[i] = nums[i], nums[nums[i] - 1]
for i in range(n):
if nums[i] != i + 1:
return i + 1
return n + 1
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/first-missing-positive/solution/que-shi-de-di-yi-ge-zheng-shu-by-leetcode-solution/
简单方法的思路:
设数组大小为N,已知最小正整数为1,则可先设缺失的最小正整数为n = 1,那么只要从1开始,判断n是否在数组里,若在数组里,则n加一,若不在数组里,则缺失的最小正整数仍为n。
但较大的数可能在较小的数之前就出现了,从前往后遍历可能会漏掉一些值,所以需要每次都从开头遍历,时间复杂度为O(N^2),不符合时间复杂度为O(n)的要求,所以要进行改进。
先进行一遍从小到大的快排,时间复杂度为O(logN),再根据前面的思路从前往后遍历数组,时间复杂度为O(N),则总时间复杂度为O(logN)+O(N)=max(O(logN),O(N))=O(N)。
class Solution:
def firstMissingPositive(self, nums: List[int]) -> int:
# 可以先加一个 set(nums) 来避免重复数
nums.sort()
n = 1
for i in nums:
if n == i:
n += 1
return n
如果显示溢出了int,则可以快排前先进行一次遍历,把负数都变成0。也可以把负数丢掉,正数移到前面去。
class Solution:
def firstMissingPositive(self, nums: List[int]) -> int:
while 0 in nums:
nums.remove(0)
nums.sort()
n = 1
for i in nums:
if n == i:
n += 1
return n
作者:everythingcanbemodeled
链接:https://leetcode-cn.com/problems/first-missing-positive/solution/yu-chu-li-kuai-pai-by-everythingcanbemod-wsfj/
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 终于写完轮子一部分:tcp代理 了,记录一下
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理