1539. 第 k 个缺失的正整数

2024-02-11 11:47:09 浏览数 (1)

链接

给你一个 严格升序排列 的正整数数组 arr 和一个整数 k

请你找到这个数组里第 k 个缺失的正整数。

示例 1:

代码语言:shell复制
输入:arr = [2,3,4,7,11], k = 5
输出:9
解释:缺失的正整数包括 [1,5,6,8,9,10,12,13,...] 。第 5 个缺失的正整数为 9 。
代码语言:shell复制
输入:arr = [1,2,3,4], k = 2
输出:6
解释:缺失的正整数包括 [5,6,7,...] 。第 2 个缺失的正整数为 6 。

prev 记录前一次访问的数,

curr 记录现在访问的数,

cnt 记录已经统计的次数,

inc 计算可以这次遍历可以增加统计的次数。

代码语言:go复制
func findKthPositive(arr []int, k int) int {
	prev, curr, cnt, inc := 0, 0, 0, 0
	for _, curr = range arr {
		inc = curr - prev - 1
		if cnt inc >= k {
			break
		}
		prev = curr
		cnt  = inc
	}
	inc = k - cnt
	return prev   inc
}

0 人点赞