LeetCode每日一题（2020.10.19） 1539. 第 k 个缺失的正整数

LeetCode每日一题（2020.10.18）

最难做的事，就是把容易做的事持之以恒。

今日题目：1539. 第 k 个缺失的正整数

难度：简单

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

通过次数5,804，提交次数10,751

这个题目的提示是存在问题的，1 <= arr.length <= 1000，当程序这样设置时，它的测试用例却超过了1000，将范围设置到3000，就可以通过测试了。先给出我自己写的代码：

class Solution:
    def findKthPositive(self, arr: List[int], k: int) -> int:
        arr1=[i for i in range(1,3000)]
        for j in range(1,3000):
            if j in arr:
                arr1.remove(j)
        if k<=len(arr1):
            return arr1[k-1]

看看运行时间和内存占用（很差）：

参考一下别人的另一种解法：

用到了昨天学习过的set()函数，上代码：

class Solution:
    def findKthPositive(self, arr: List[int], k: int) -> int:
        a=[x for x in range(1,len(arr)+k+1)]
        b=set(a)-set(arr)
        bl=list(b)
        bl.sort()
        return bl[k-1]

他这里的循环次数是根据k值变化的，从而减少了固定循环次数的时间浪费，以后在自己写程序的时候也可以做这样的处理，以减少循环次数。

然后利用set()函数的差运算，找出两个之间的差集，在用list()函数转化为列表，再对列表进行排序（默认从小到大），然后访问下标为k-1的值，方便快捷。看下优化后的运行时间：

还可以做进一步的优化，将循环次数len(arr)+k+1，改为arr[-1]+k+1，运行时间也会减少一些：

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