몰?.루();

백준 1715 코틀린 본문

프로그래밍/안드로이드, 코틀린

백준 1715 코틀린

toonraon 2022. 9. 29. 00:11
import java.util.PriorityQueue

fun main() {
    val n = readln().toInt()
    val arr = PriorityQueue<Long>()

    repeat(n) {
        arr.add(readln().toLong())
    }

    var answer = 0L
    while (arr.size > 1) {
        val a = arr.poll()
        val b = arr.poll()
        val s = a + b
        arr.add(s)
        answer += s
    }

    println(answer)
}

무조건 오름차순으로 정렬된 배열이 필요할 땐 PriorityQueue를 쓰는 게 좋다는 걸 배울 수 있었던 문제입니다.

이걸 몰라서 괜히 LinkedList 쓰고 binarySearch 쓰고 별 쑈를 다 했네...

Comments