每日一题-第 k 个缺失的正整数

---

题目

给你一个 严格升序排列 的正整数数组 arr 和一个整数 k 。

请你找到这个数组里第 k 个缺失的正整数。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/kth-missing-positive-number
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

示例

示例 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。

提示：

1 <= arr.length <= 1000
1 <= arr[i] <= 1000
1 <= k <= 1000
对于所有 1 <= i < j <= arr.length 的 i 和 j 满足 arr[i] < arr[j]

解题

遍历列表，将元素未出现的计数，计数与k相等时，返回当前元素；如果将arr遍历完了也没有出现，那么直接返回arr最后一个元素加上剩余未计数的个数。

class Solution:
    def findKthPositive(self, arr, k):
        count = 0
        for i in range(1, arr[-1]):
            if i not in arr:
                count += 1
                if count == k:
                    return i
        return k - count + arr[-1]