Baekjoon(Kotlin) - 11279. 최대 힙

URL : https://www.acmicpc.net/problem/11279

11279번: 최대 힙

 

[ RESULT ]

 

import java.io.BufferedReader
import java.io.BufferedWriter
import java.io.InputStreamReader
import java.io.OutputStreamWriter
import java.util.PriorityQueue

fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val bw = BufferedWriter(OutputStreamWriter(System.out))

    val N:Int = br.readLine().toInt()
    var data: Int
    val maxHeap = PriorityQueue<Int>(reverseOrder())

    repeat(N) {
        data = br.readLine().toInt()
        if(data == 0) {
            if(maxHeap.size == 0) bw.write("0\n").toString()
            else {
                val max = maxHeap.poll()
                bw.write("$max\n")
            }
        }
        else maxHeap.add(data)
    }


    br.close()
    bw.flush()
    bw.close()
}

 

 

최대 힙을 어떻게 구현할지 구상하던 중, Kotlin에 PriorityQueue 를 이용하여 쉽게 최대/최소 힙을 구성할 수 있다는 것을 알고 이를 이용하여 코들르 작성했다.