[数据结构与算法]力扣 1539. 第 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]

解释与代码

可以发现arr[i] - i - 1就是前面缺失的数字的数量,可以使用二分来写

class Solution {
public:
    int findKthPositive(vector<int>& arr, int k) {
        if (arr[0] > k) return k;
        int l = 0, r = arr.size() - 1;
        while (l < r) {
            int mid = (l + r + 1) >> 1;
            if (arr[mid] - mid - 1 < k) {
                l = mid;
            } else {
                r = mid - 1;
            }
        }
        return arr[l] + (k - arr[l] + l + 1);
    }
};