[백준] 2805번: 나무 자르기 - 코틀린(kotlin)

https://www.acmicpc.net/problem/2805

2805번: 나무 자르기

 

문제 해석

나무가 M미터 필요하다.
이때 나무를 자르기 위해 절단기의 높이 H를 설정해야 한다.
나무를 최소한으로 필요한 만큼만 가져가려고 합니다.

 

코드

fun main() {
    val (hasTreeCount, needTreeCount) = readLine()!!.split(" ").map { it.toInt() }
    val treeHeights = IntArray(hasTreeCount)
    var index = 0
    readLine()!!.split(" ").map { treeHeights[index++] = it.toInt() }

    val maxHeight = treeHeights.maxOrNull()!!

    var left = 0
    var right = maxHeight
    var answer = 0L

    while (left <= right) {
        val middle = (left + right) / 2
        var curHasTreeCount = 0L
        for (curTree in 0 until hasTreeCount) {
            if (treeHeights[curTree] > middle) {
                curHasTreeCount += treeHeights[curTree].toLong() - middle.toLong()
            }
        }

        if (curHasTreeCount >= needTreeCount) {
            answer = middle.toLong()
            left = middle + 1
        } else {
            right = middle - 1
        }
    }
    println(answer)
}