[백준] 생태학 (4358)(kotlin)

원본 문제

문제 설명

생태학에서 나무의 분포도를 측정하는 것은 중요하다. 그러므로 당신은 미국 전역의 나무들이 주어졌을 때, 각 종이 전체에서 몇 %를 차지하는지 구하는 프로그램을 만들어야 한다.

입력

프로그램은 여러 줄로 이루어져 있으며, 한 줄에 하나의 나무 종 이름이 주어진다. 어떤 종 이름도 30글자를 넘지 않으며, 입력에는 최대 10,000개의 종이 주어지고 최대 1,000,000그루의 나무가 주어진다.

출력

주어진 각 종의 이름을 사전순으로 출력하고, 그 종이 차지하는 비율을 백분율로 소수점 4째자리까지 함께 출력한다.

테스트 케이스

입력	출력
Red Alder Ash Aspen Basswood Ash Beech Yellow Birch Ash Cherry Cottonwood Ash Cypress Red Elm Gum Hackberry White Oak Hickory Pecan Hard Maple White Oak Soft Maple Red Oak Red Oak White Oak Poplan Sassafras Sycamore Black Walnut Willow	Ash 13.7931 Aspen 3.4483 Basswood 3.4483 Beech 3.4483 Black Walnut 3.4483 Cherry 3.4483 Cottonwood 3.4483 Cypress 3.4483 Gum 3.4483 Hackberry 3.4483 Hard Maple 3.4483 Hickory 3.4483 Pecan 3.4483 Poplan 3.4483 Red Alder 3.4483 Red Elm 3.4483 Red Oak 6.8966 Sassafras 3.4483 Soft Maple 3.4483 Sycamore 3.4483 White Oak 10.3448 Willow 3.4483 Yellow Birch 3.4483

문제 풀이1(Trie구조)

data class TrieNode(var child: MutableMap<Char, TrieNode>, var count: Int = 0, var terminal: Boolean = false, var name: String = "")

class Trie(private val root: TrieNode = TrieNode(sortedMapOf(), 0)) {
    var total: Int = 0

    fun insert(tree: String) {
        var currentNode = root
        for (char in tree) {
            currentNode = currentNode.child.getOrPut(char) { TrieNode(sortedMapOf(), 0)}
        }

        if (currentNode.terminal) {
            currentNode.count++
            total++
        }
        else {
            total++
            currentNode.terminal = true
            currentNode.count = 1
        }
    }

    fun findAll() {
        for (node in root.child) {
            dfs(node.value, node.key.toString())
        }
    }

    private fun dfs(nodes: TrieNode, name: String) {
        if (nodes.terminal) println("${name} ${String.format("%.4f", (nodes.count/total.toFloat())*100)}")
        else {
            for (node in nodes.child) {
                dfs(node.value, name + node.key)
            }
        }
    }
}

fun main() = with(System.`in`.bufferedReader()) {
    var n = readLine()
    val trie = Trie()
    while (n != null) {
        trie.insert(n)
        n = readLine()
    }
    trie.findAll()
}