DNPotapov / Leetcode

0 stars 0 forks source link

1539. Kth Missing Positive Number #10

Open DNPotapov opened 1 year ago

DNPotapov commented 1 year ago
class Solution(object):
    def findKthPositive(self, arr, k):
        """
        :type arr: List[int]
        :type k: int
        :rtype: int
        """
        res = []

        for i in range(1, max(arr)):
            if i not in arr and len(res) != k:
                res.append(i)
        if not res:
            return max(arr) + k
        if len(res) == k:
            return max(res)
        else:
            return (max(arr) + (k - len(res)))
DNPotapov commented 1 year ago

https://leetcode.com/problems/kth-missing-positive-number/