【Leetcode】1539. Kth Missing Positive Number

题目地址：

https://leetcode.com/problems/kth-missing-positive-number/

给定一个正整数严格递增长$n$的数组$A$，再给定一个正整数$k$，问$A$里缺的第$k$个正整数是多少。例如$A=(2,3,4,7)$，$k=5$，那么$A$缺的正整数是$(1,5,6,8,9)$，第$5$个是$9$。

思路是二分。对于某个$i$，我们考虑$1\sim A[i]$中的$A$所缺的数字个数，那么显然缺的个数是$f[i]=A[i]-(i+1)$，并且$f$是非降的。我们可以找到最大的使得$f[i]<k$的$i$，这样所缺的第$k$个数字一定位于$A[i]+1,...,A[i+1]$之中，实际上就是$k-(A[i]-(i+1))+A[i]=k+i+1$；如果不存在使得$f[i]<k$的$i$，说明$f[0]\ge k$了，那么缺的数就是$k$。代码如下：

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

    return a[l] - (l + 1) >= k ? k : k + l + 1;
  }
};

时间复杂度$O(\log n)$，空间$O(1)$。