[Kotlin] 백준 - 빈도 정렬

https://www.acmicpc.net/problem/2910

[문제]

위대한 해커 창영이는 모든 암호를 깨는 방법을 발견했다. 그 방법은 빈도를 조사하는 것이다.

창영이는 말할 수 없는 방법을 이용해서 현우가 강산이에게 보내는 메시지를 획득했다. 이 메시지는 숫자 N개로 이루어진 수열이고, 숫자는 모두 C보다 작거나 같다. 창영이는 이 숫자를 자주 등장하는 빈도순대로 정렬하려고 한다.

만약, 수열의 두 수 X와 Y가 있을 때, X가 Y보다 수열에서 많이 등장하는 경우에는 X가 Y보다 앞에 있어야 한다. 만약, 등장하는 횟수가 같다면, 먼저 나온 것이 앞에 있어야 한다.

이렇게 정렬하는 방법을 빈도 정렬이라고 한다.

수열이 주어졌을 때, 빈도 정렬을 하는 프로그램을 작성하시오.

첫째 줄에 메시지의 길이 N과 C가 주어진다. (1 ≤ N ≤ 1,000, 1 ≤ C ≤ 1,000,000,000)

둘째 줄에 메시지 수열이 주어진다.

첫째 줄에 입력으로 주어진 수열을 빈도 정렬한 다음 출력한다.

 

[입출력 예]

 

[풀이]

먼저 각 숫자를 키로 생각하여 하나의 클래스로 정리합니다.

실제 값 key, 최초 등장 index, 총 등장 횟수 count

data class Message(
    val key: Int,
    var index: Int,
    var count: Int
)

 

입력을 받은 뒤, 같은 키 값에 대해서 데이터를 업데이트해주어야 합니다.

inputs.forEachIndexed { index, input ->
    val exist = temps.find { it.key == input } //같은 키 찾기
    if (exist == null) {
        temps.add(Message(input, index, 1))
    } else {
        temps.remove(exist)
        temps.add(Message(input, exist.index, exist.count + 1))
    }
}

리스트에 이미 값이 존재한다면 기존 값 기반으로 새롭게 업데이트해주고, 없다면 추가해 줍니다.

 

이제 문제의 내용에 따라 Message 리스트를 정렬해 줍니다.

 val sortedTemps = temps.sortedWith(
        compareByDescending<Message> { it.count }
            .thenBy { it.index }
    )

먼저 횟수가 큰 것부터 나오도록 count를 기준으로 내림차순 정렬해 주고, count가 같다면 최초 등장 index를 기준으로 오름차순 정렬해 주면 됩니다.

 

아래는 코틀린 전체 코드입니다.

data class Message(
    val key: Int,
    var index: Int,
    var count: Int
)

fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val counts = br.readLine()
    val inputs = br.readLine().split(" ").map { it.toInt() }
    val temps = mutableListOf<Message>()

    inputs.forEachIndexed { index, input ->
        val exist = temps.find { it.key == input } //같은 키 찾기
        if (exist == null) {
            temps.add(Message(input, index, 1))
        } else {
            temps.remove(exist)
            temps.add(Message(input, exist.index, exist.count + 1))
        }
    }
    val sortedTemps = temps.sortedWith(
        compareByDescending<Message> { it.count }
            .thenBy { it.index }
    )
    val result = StringBuilder()
    sortedTemps.forEach { value ->
        repeat(value.count) {
            result.append("${value.key} ")
        }
    }
    println(result.toString())
}