LeetCode|1539.第k个缺失的正整数

LeetCode习题答案汇总

题目:
给你一个 严格升序排列 的正整数数组 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 。

提示:

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

分析:

  • 定义一个数组nums,长度为arr中最大的数再加上k+1
  • 遍历arr,将nums中出现的arr中的数字置零
  • 输出计数等于k时候nums对应的数

代码:

class Solution:
    def findKthPositive(self, arr: List[int], k: int) -> int:
        nums = [i for i in range(arr[-1]+k+1)]
        count = 0
        for i in arr:
            nums[i] = 0
        for i in range(1,len(nums)):
            if nums[i] != 0:
                count += 1
            if count == k:
                return nums[i]

结果:
LeetCode|1539.第k个缺失的正整数_第1张图片

你可能感兴趣的:(LeetCode)