[백준 - 5639] 이진 검색 트리

문제링크

import java.io.BufferedWriter

fun main() {
    val bufferedReader = System.`in`.bufferedReader()
    val bufferedWriter = System.out.bufferedWriter()

    val preOrder = mutableListOf<Int>()
    while (true) {
        val node = bufferedReader.readLine() ?: break
        preOrder.add(node.toInt())
    }

    getPostOrder(preOrder, 0, preOrder.size - 1, bufferedWriter)

    bufferedReader.close()
    bufferedWriter.close()
}

fun getPostOrder(preOrder: List<Int>, left: Int, right: Int, bufferedWriter: BufferedWriter) {
    if (left > right) return

    val root = preOrder[left]
    var rightChildIndex = -1
    for (i in (left + 1)..right) {
        if (root < preOrder[i]) {
            rightChildIndex = i
            break
        }
    }
    if (rightChildIndex == -1) {
        rightChildIndex = right + 1
    }
    getPostOrder(preOrder, left + 1, rightChildIndex - 1, bufferedWriter)
    getPostOrder(preOrder, rightChildIndex, right, bufferedWriter)
    bufferedWriter.write("$root\n")
}

주석 없는 코드를 만들기 위해 노력하는 개발자입니다.

혹시라도 의도가 분명하지 않아보이는 (이해가 되지 않는) 코드가 있으시다면 편하게 답변 달아주시면 정말 감사하겠습니다.

reference

• https://8iggy.tistory.com/112