题目地址:
https://leetcode.com/problems/kth-missing-positive-number/
给定一个正整数严格递增长 n n n的数组 A A A,再给定一个正整数 k k k,问 A A A里缺的第 k k k个正整数是多少。例如 A = ( 2 , 3 , 4 , 7 ) A=(2,3,4,7) A=(2,3,4,7), k = 5 k=5 k=5,那么 A A A缺的正整数是 ( 1 , 5 , 6 , 8 , 9 ) (1,5,6,8,9) (1,5,6,8,9),第 5 5 5个是 9 9 9。
思路是二分。对于某个 i i i,我们考虑 1 ∼ A [ i ] 1\sim A[i] 1∼A[i]中的 A A A所缺的数字个数,那么显然缺的个数是 f [ i ] = A [ i ] − ( i + 1 ) f[i]=A[i]-(i+1) f[i]=A[i]−(i+1),并且 f f f是非降的。我们可以找到最大的使得 f [ i ] < k f[i]<k f[i]<k的 i i i,这样所缺的第 k k k个数字一定位于 A [ i ] + 1 , . . . , A [ i + 1 ] A[i]+1,...,A[i+1] A[i]+1,...,A[i+1]之中,实际上就是 k − ( A [ i ] − ( i + 1 ) ) + A [ i ] = k + i + 1 k-(A[i]-(i+1))+A[i]=k+i+1 k−(A[i]−(i+1))+A[i]=k+i+1;如果不存在使得 f [ i ] < k f[i]<k f[i]<k的 i i i,说明 f [ 0 ] ≥ k f[0]\ge k f[0]≥k了,那么缺的数就是 k 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(\log n) O(logn),空间 O ( 1 ) O(1) O(1)。