leetcode-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] 

解法一：（枚举）

class Solution:
    def findKthPositive(self, arr: List[int], k: int) -> int:
        count = i = missnum = 0
        current = 1
        while count < k:
            if arr[i] == current:
                if i + 1 < len(arr):
                    i += 1
            else:
                count += 1
                missnum = current
            current += 1
        return missnum

解法二：（二分法）

class Solution:
    def findKthPositive(self, arr: List[int], k: int) -> int:
        if arr[0] > k:
            return k
        
        l, r = 0, len(arr)
        while l < r:
            mid = (l + r) >> 1
            x = arr[mid] if mid < len(arr) else 10**9
            if x - mid - 1 >= k:
                r = mid
            else:
                l = mid + 1

        return k - (arr[l - 1] - (l - 1) - 1) + arr[l - 1]

解法三：（集合相减）

class Solution:
    def findKthPositive(self, arr: List[int], k: int) -> int:
        return sorted(set(range(1,max(arr) + k + 1)) - set(arr))[k-1]

解法四：（列表遍历比较）

class Solution:
    def findKthPositive(self, arr: List[int], k: int) -> int:
        lis = [i for i in range(1, max(arr)+k+1)]
        for a in arr:
            lis.remove(a)
        return lis[k-1]

解法五：（简化二分法）

class Solution:
    def findKthPositive(self, arr: List[int], k: int) -> int:
        if k < arr[0]:
            return k
        
        l, r = 0, len(arr)-1
        while l <= r:
            mid = (l + r) // 2
            
            if arr[mid] - mid > k:
                r = mid - 1
            else:
                l = mid + 1
        return k + l