leetcode 41. 缺失的第一个正数

题解

缺失的第一个正数不会大于 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
}