- 第 k 个缺失的正整数
给你一个 严格升序排列 的正整数数组 arr 和一个整数 k 。
请你找到这个数组里第 k 个缺失的正整数。
示例 1:
输入:arr = [2,3,4,7,11], k = 5
输出:9
解释:缺失的正整数包括 [1,5,6,8,9,10,12,13,…] 。第 5 个缺失的正整数为 9 。
思路
已知的一个规律是缺省的元素的值=k+已有的元素的个数。
缺的元素个数有这么一个规律,arr[i]-i-1,即当前的值减去已有的元素个数,所以我们可以判断出缺省的元素数量。那么找到离k最近的缺省元素的个数,就是已有的元素的个数。这个最近的位置就应该是arr[i]-i-1<k。这个就是我们用于二分的条件。
class Solution {
public:
int findKthPositive(vector<int>& arr, int k) {
if(arr[0]>k)
{
return k;
}
int left=0,right=arr.size()-1;
while(left<right)
{
int mid=left+(right-left)/2;
if(arr[mid]-mid-1<k)
{
left=mid+1;
//找到比k小的第一个元素
if(arr[left]-mid-1>k)
{
left--;
break;
}
}else
{
right=mid-1;
}
}
return k+left+1;
}
};