[백준/코틀린] 18111번: 마인크래프트

링크

18111번: 마인크래프트

풀이

땅을 특정 높이 h로 고르기 위해서 필요한 블록 제거량(remove)과 추가량(add)을 계산합니다.
현재 가지고 있는 블록 수(b)에 제거량을 더한 값이 추가량 이상이라면, 해당 높이를 만들 수 있습니다.

모든 가능한 높이(0~256)에 대해, 해당 높이로 ‘땅 고르기’가 가능한지 확인하고,
그중 가장 적은 시간이 걸리는 높이와 그 시간을 구합니다.

코드

fun main() {
    val (n, m, b) = readln().split(" ").map { it.toInt() }
    val field = List(n) { readln().split(" ").map { it.toInt() } }.flatten()
    var minTime = Int.MAX_VALUE
    var maxHeight = -1

    for (height in 0..256) {
        var remove = 0
        var add = 0

        field.forEach {
            when {
                height < it -> remove += (it - height)
                it < height -> add += (height - it)
            }
        }

        if (b + remove >= add) {
            val time = 2 * remove + add

            if (time <= minTime) {
                minTime = time
                maxHeight = height
            }
        }
    }

    println("$minTime $maxHeight")
}