leetcode--python--1539

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)