Coding Test

[Kotlin] 백준 - 역원소 정렬

SeungYong.Lee 2024. 9. 24. 21:36
반응형

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

모든 원소가 양의 정수인 집합이 있을 때, 원소를 거꾸로 뒤집고 그 원소를 오름차순으로 정렬하는 프로그램을 작성하세요.

단, 원소를 뒤집었을 때 0이 앞에 선행되는 경우는 0을 생략해야 합니다.

첫 번째로 입력되는 건 n (1 ≤ n ≤ 106)으로 사용자가 뒤이어 입력할 원소값을 결정합니다. 입력하는 줄에는 하나의 원소값뿐만 아니라 여러 원소값도 들어갈 수 있습니다.

단, 입력하는 정수는 1012를 넘어선 안 됩니다.

출력문은 위 문제 내용에 나와있는 정렬방법으로 정렬하여 아래 예제 출력을 참고하여 출력하세요.

 

[풀이]

문제 자체는 쉽게 이해할 수 있는데, 자잘한 오류가 발생하는 구간들이 있네요.

일단 첫 번째로 입력되는 건 사용자가 입력할 원소 개수 값들을 말하는 것 같습니다. 이 부분 유의해주세요.

fun main() {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val nums = mutableListOf<String>()
    while (true) {
        val input = br.readLine()
        if (input.isNullOrEmpty()) break
        nums.addAll(input.split(" "))
    }
    
    val reversedNumbers = nums.map {
        it.reversed().toInt()
    }.sorted()  

    reversedNumbers.forEach { println(it) }
}

입력 값들을 모두 리스트에 넣어주고, 다시 리스트를 돌면서 각 원소를 reverse 시킨 후, int형으로 변경해 줍니다.

이후 전체 리스트를 오름 차순 처리해 줍니다.

 

아래와 같은 풀이도 가능합니다.

fun main() = with(System.`in`.bufferedReader()) {
    val list = ArrayList<String>()
    var str = readLine()

    while (str != null) {
        val st = StringTokenizer(str)
        while (st.hasMoreTokens()) list.add(st.nextToken())
        str = readLine()
    }
    list.removeAt(0)

    val sb = StringBuilder()
    list.map { it.reversed().toLong() }.sorted().forEach { sb.append(it).append("\n") }
    print(sb)
}
반응형