1539: 第 k 个缺失的正整数
给你一个 严格升序排列 的正整数数组 arr 和一个整数 k 。
请你找到这个数组里第 k 个缺失的正整数。
class Solution(object):
def findKthPositive(self, arr, k):
"""
:type arr: List[int]
:type k: int
:rtype: int
"""
end = arr[-1]
length = len(arr)
if end - length < k:
result = end + k - (end - length)
else:
list_end = list(range(1, end+1))
result_list = sorted(list(set(list_end) - set(arr)))
result = result_list[k-1]
return(result)
首先判定arr数组最后一个元素,用来和数组总长度作比较,若这个数减去k大于len(arr),证明第k个缺失的数在list(range(1,末尾元素))内,那就可以通过数组相减得到缺失值数组,然后在该数组访问第k-1个下标即为所求;否则,说明缺失的值超出了arr末尾元素的值,那么最后结果就等于上面代码中的result = end + k - (end - length)