HackerRank/Algorithm/JesseAndCookies kotlin

JesseAndCookies.kt

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

private fun cookies(
    sweetness: Int,
    cookies: Array<Int>,
): Int {
    val result = PriorityQueue<Int>().apply { addAll(cookies) }
    var countOfIteration = 0
    while (result.peek() < sweetness && result.size > 1) {
        val sweetenCookie = result.poll() + result.poll() * 2
        result.add(sweetenCookie)
        countOfIteration++
    }
    return if (result.peek() < sweetness) {
        -1
    } else {
        countOfIteration
    }
}

// if you use this extension, one of the test cases will throw timeout error.
private fun PriorityQueue<Int>.checkIfSweetenNeeded(sweetness: Int): Boolean =
    this.peek() < sweetness && this.size > 1

fun main(args: Array<String>) {
    val scanner = Scanner(System.`in`)
    val (_, sweetness) = scanner.nextLine().trimEnd()
        .split(" ")
        .map(String::toInt)

    val cookies = scanner.nextLine().trimEnd()
        .split(" ")
        .map(String::toInt)
        .toTypedArray()

    val result = cookies(sweetness, cookies)

    println(result)
}