LeetCode in Kotlin

1642. Furthest Building You Can Reach

Medium

You are given an integer array heights representing the heights of buildings, some bricks, and some ladders.

You start your journey from building 0 and move to the next building by possibly using bricks or ladders.

While moving from building i to building i+1 (0-indexed),

Return the furthest building index (0-indexed) you can reach if you use the given ladders and bricks optimally.

Example 1:

Input: heights = [4,2,7,6,9,14,12], bricks = 5, ladders = 1

Output: 4

Explanation: Starting at building 0, you can follow these steps:

It is impossible to go beyond building 4 because you do not have any more bricks or ladders.

Example 2:

Input: heights = [4,12,2,7,3,18,20,3,19], bricks = 10, ladders = 2

Output: 7

Example 3:

Input: heights = [14,3,19,3], bricks = 17, ladders = 0

Output: 3

Constraints:

Solution

import java.util.PriorityQueue

@Suppress("NAME_SHADOWING")
class Solution {
    fun furthestBuilding(heights: IntArray, bricks: Int, ladders: Int): Int {
        var bricks = bricks
        val minHeap = PriorityQueue<Int>()
        var i = 0
        // we'll assume to use ladders for the first l jumps and adjust it afterwards
        while (i < heights.size - 1 && minHeap.size < ladders) {
            val diff = heights[i + 1] - heights[i]
            if (diff > 0) {
                minHeap.offer(diff)
            }
            i++
        }
        while (i < heights.size - 1) {
            val diff = heights[i + 1] - heights[i]
            if (diff > 0) {
                if (minHeap.isNotEmpty() && minHeap.peek() < diff) {
                    bricks -= minHeap.poll()
                    minHeap.offer(diff)
                } else {
                    bricks -= diff
                }
                if (bricks < 0) {
                    return i
                }
            }
            i++
        }
        return i
    }
}