[백준 1431번] [Kotlin] 시리얼 번호

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

1431번: 시리얼 번호

 

난이도 : 실버 3
태그 : 정렬

 

 

설명

1. A와 B의 길이가 다르면, 짧은 것이 먼저 온다.
2. 만약 서로 길이가 같다면, A의 모든 자리수의 합과 B의 모든 자리수의 합을 비교해서 작은 합을 가지는 것이 먼저온다. (숫자인 것만 더한다)
3. 만약 1,2번 둘 조건으로도 비교할 수 없으면, 사전순으로 비교한다. 숫자가 알파벳보다 사전순으로 작다.

 

다소 복잡한 정렬조건을 갖고 푸는 문제입니다.

하지만, 다중 속성 정렬 기능을 사용하면 간단히 풀 수 있습니다.

 

 

 

소스코드

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

    val n = br.readLine().toInt()
    val texts = Array(n) { "" }

    repeat(n) {
        texts[it] = br.readLine()
    }

    texts.sortWith(compareBy(
        { it.length },
        { it.filter { it2 -> it2.isDigit() }.sumOf { it3 -> it3.digitToInt() } },
        { it }
    ))

    texts.forEach { println(it) }
}

 

 

sortWith와 compareBy를 사용해 다중 조건으로 정렬할 수 있습니다.

 

한 조건씩 보겠습니다.

 

 

1번 조건 (길이로 정렬)

 

it.length로 문자열의 길이를 기준으로 정렬할 수 있습니다.

 

 

 

 

2번 조건 (숫자들의 합으로 정렬)

 

filter를 사용해 문자열에서 특정 조건을 만족하는 문자만 뽑아와 새 문자열을 만들 수 있습니다.

위 코드에서는 Char의 메소드중 하나인 숫자인지 확인하는 메소드인 isDigit()을 사용해 숫자일 경우만 뽑아왔습니다.

 

그리고 sumOf를 사용해 특정 조건에 해당하는 값의 합을 구하거나, 문자열의 각 원소를 가공하여 합을 구할 수 있습니다.

위 코드에서는 직전에서 filter로 return 받은 문자열의 각각의 문자(char)에 접근하여

 

digitToInt()로 char 형태의 값을 Int 형태로 가공하여 합을 구합니다.

toInt()를 사용해도 동일한 값이 나오나,

String.toInt()와 달리 Char.toInt()는 deprecated 처리 되었기에 사용하는 것을 권장하진 않습니다.

 

 

 

3번 조건 (사전순 정렬)

 

마지막으로 사전순 정렬은 문자열 그 자체 (it)을 넣으면 됩니다.

 

 

 

후기

코틀린의 람다식을 활용해 풀이하였습니다.

람다식이 처음엔 이해하기는 어렵긴 하지만,

이해하면 굉장히 간단하게 코드를 짤 수 있는 것 같습니다.