Условие

Дан отсортированный по возрастанию массив положительных целых arr и целое k. Требуется вернуть k-е пропущенное положительное целое число вне этого массива.

Решение

class Solution {
    fun findKthPositive(arr: IntArray, k: Int): Int {
        var missing = k
        var prev = 0
        for (num in arr) {
            val gap = num - prev - 1
            if (missing <= gap) return prev + missing
            missing -= gap
            prev = num
        }
        return prev + missing
    }
}