LeetCode 41. 缺失的第一个正数
给你一个未排序的整数数组,请你找出其中没有出现的最小的正整数。
示例 1:
输入: [1,2,0]
输出: 3
示例 2:
输入: [3,4,-1,1]
输出: 2
示例 3:
输入: [7,8,9,11,12]
输出: 1
算法的时间复杂度应为O(n),并且只能使用常数级别的额外空间。
思路一:可以建一个hash表,存储出现的正数,然后再进行遍历找出缺失值,但是需要花费额外的O(N)空间。
class Solution:
def firstMissingPositive(self, nums: List[int]) -> int:
hashmap = defaultdict(int)
if not nums: # 数组为空,返回1
return 1
for num in nums:
hashmap[num] = num
for i in range(1, len(nums) + 1):
if i not in hashmap:
return i # 不存在,则返回
return len(nums) + 1 # 都存在,返回下一个正数
思路二:首先新建一个数组,用来存放值为0到N的数,
遍历数组把数组中大于0并且小于等于数组长度的值放入对应的位置。
遍历新数组,如果该位置为空,那么就是缺失的数。
class Solution:
def firstMissingPositive(self, nums: List[int]) -> int:
n = len(nums)
dp = [-1] * (n + 1)
for i in range(n):
if nums[i] > 0 and nums[i] <= n:
dp[nums[i]] = nums[i]
for i in range(1, n + 1):
if dp[i] == -1:
return i
return n + 1
这种解法时间复杂度满足要求,但是开辟了N的空间,不符合要求。可以进行优化。
可以不使用dp数组,直接在原始数组中操作,把每个数放到正确的位置。
从第一个位置开始,让座位上的乘客走到自己应该坐的位置,并让该位置的人坐到第一个位置。一直进行这样的操作,直到第一个位置的人坐到自己位置。不过有时候我们知道,有的人总喜欢逃票。因此终止条件就是,一旦发现第一个位置的人逃票(票号 <= 0,或 >= 最大座位号),则终止交换。
对第二到N个位置做相同的操作。
class Solution:
def firstMissingPositive(self, nums: List[int]) -> int:
# n = len(nums)
# dp = [-1] * (n + 1)
# for i in range(n):
# if nums[i] > 0 and nums[i] <= n:
# dp[nums[i]] = nums[i]
# for i in range(1, n + 1):
# if dp[i] == -1:
# return i
# return n + 1
n = len(nums)
for i in range(1, n):
# 数字nums[i]应该在nums[i]-1这个位置,例如,1应该在第0的位置
while 0 < nums[i] and nums[i] <= n and nums[i] != nums[nums[i] - 1]:
# 和坐在自己应该坐的位置上的人换位置
value = nums[i]
ori_value = nums[value - 1]
nums[value - 1] = value # nums[i]走到正确的位置(value-1)
nums[i] = ori_value # 原来位置上的人坐到nums[i]的位置上(i)
# nums[i], nums[nums[i] - 1] = nums[nums[i] - 1], nums[i] 会改变值
for i in range(n):
if nums[i] != i + 1:
return i + 1
return n + 1