給你一個 嚴格升序排列 的正整數(shù)數(shù)組 arr 和一個整數(shù) k 。
請你找到這個數(shù)組里第 k 個缺失的正整數(shù)。
示例 1:
輸入:arr = [2,3,4,7,11], k = 5
輸出:9
解釋:缺失的正整數(shù)包括 [1,5,6,8,9,10,12,13,...] 。第 5 個缺失的正整數(shù)為 9 。
示例 2:
輸入:arr = [1,2,3,4], k = 2
輸出:6
解釋:缺失的正整數(shù)包括 [5,6,7,...] 。第 2 個缺失的正整數(shù)為 6 。
提示:
- 1 <= arr.length <= 1000
- 1 <= arr[i] <= 1000
- 1 <= k <= 1000
- 對于所有 1 <= i < j <= arr.length 的 i 和 j 滿足 arr[i] < arr[j]
我的算法實現(xiàn)為:
/**
* @param {number[]} arr
* @param {number} k
* @return {number}
*/
var findKthPositive = function (arr, k) {
// 如果數(shù)組的最大值 - 數(shù)組長度 >= k ,那么直接返回最后一個加上最后的差值
const len = arr.length
const lastMax = arr[len - 1]
if (lastMax - len < k) {
return lastMax + k - (lastMax - len)
}
let n
for (let i = 1, j = 0; k > 0; i++) {
// 如果不相等,說明這個元素沒有
if (arr[j] !== i) {
// 如果 k 到 0 ,就說明到頭了
k--
// 記錄被缺的這個值
n = i
continue
}
j++
}
return n
};
來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/kth-missing-positive-number
著作權歸領扣網(wǎng)絡所有。商業(yè)轉(zhuǎn)載請聯(lián)系官方授權,非商業(yè)轉(zhuǎn)載請注明出處。