HackerRank/Algorithm/JesseAndCookies kotlin2

JesseAndCookies2.kt

import java.util.*
import kotlin.*
import kotlin.collections.*

private fun List<Int>.sweetenCookies(targetSweetness: Int): Int {
    val heap = PriorityQueue(this)

    var count = 0
    while (heap.size > 1 && heap.peek() <= targetSweetness) {
        val first = heap.poll()
        val second = heap.poll()
        heap.add(first + second * 2)
        count++
    }

    return if (heap.peek() < targetSweetness) {
        -1
    } else {
        count
    }
}

fun main(args: Array<String>) {
    val (_, sweetness) = readln().trimEnd()
        .split(" ")
        .map(String::toInt)

    val cookies = readln().trimEnd()
        .split(" ")
        .map(String::toInt)

    val result = cookies.sweetenCookies(targetSweetness = sweetness)

    println(result)
}