leetcode 41. 缺失的第一个正数

题目描述

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

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

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

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

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

题解

缺失的第一个正数不会大于 n+1;首先将小于 1 的值置为 n+1,然后数组的值作为下标,把相对应的位置 nums[v-1] 变为负数,最后遍历数组判断是否有正数

示例代码(go)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
func firstMissingPositive(nums []int) int {
n := len(nums)
for i, v := range nums {
if v < 1 {
nums[i] = n+1
}
}
for _, v := range nums {
v1 := abs(v)
if v1 > 0 && v1 <= n {
nums[v1-1] = -abs(nums[v1-1])
}
}
for i, v := range nums {
if v > 0 {
return i+1
}
}
return n+1
}

func abs(a int) int {
if a < 0 {
return -a
}
return a
}