LeetCode 第 k 个缺失的正整数

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

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

示例 1：

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

示例 2：

输入：arr = [1,2,3,4], k = 2
输出：6
解释：缺失的正整数包括 [5,6,7,...] 。第 2 个缺失的正整数为 6 。

提示：

• 1 <= arr.length <= 1000
• 1 <= arr[i] <= 1000
• 1 <= k <= 1000
• 对于所有 1 <= i < j <= arr.length 的 i 和 j 满足 arr[i] < arr[j]

我的算法实现为：

/**
 * @param {number[]} arr
 * @param {number} k
 * @return {number}
 */
    var findKthPositive = function (arr, k) {
      // 如果数组的最大值 - 数组长度 >= 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。