【题目描述】
给你一个 严格升序排列 的正整数数组 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 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/kth-missing-positive-number
【解题思路】
如果是严格不缺失元素的数组,会有arr[i]=i+1,但现在有缺失,缺失的元素的数量为arr[i]-(i+1),所以要查出第k个缺失的元素,就要找到k=arr[i]-(i+1),又因为数组是单调递增的,所以可以用二分法来解决这个问题,可以找到缺失的元素个数k=arr[i]-i-1,则缺失的第k个元素为找到的这个位置加上k。用Python实现的代码如下:
class Solution(object):
def findKthPositive(self, arr, k):
"""
:type arr: List[int]
:type k: int
:rtype: int
"""
left, right=0, len(arr)
while left<right:
mid = (left+right)/2
if k <= arr[mid]-mid-1:
right=mid
else:
left=mid+1
return k+left