Условие

Дан массив nums, состоящий из 0 и 1, и целое число k. Необходимо проверить, находятся ли все 1 в массиве на расстоянии не менее k позиций друг от друга. Вернуть true, если условие выполнено, и false в противном случае.

Решение

class Solution {
    fun kLengthApart(nums: IntArray, k: Int): Boolean {
        var prevOne = -k - 1
        for (i in nums.indices) {
            if (nums[i] == 1) {
                if (i - prevOne - 1 < k) {
                    return false
                }
                prevOne = i
            }
        }
        return true
    }
}