[백준] 시리얼 번호 (1431)(kotlin)
문제 설명
입력 및 출력
» 입력
첫째 줄에 기타의 개수 N이 주어진다. N은 1,000보다 작거나 같다. 둘째 줄부터 N개의 줄에 시리얼 번호가 하나씩 주어진다. 시리얼 번호의 길이는 최대 50이고, 알파벳 대문자 또는 숫자로만 이루어져 있다. 시리얼 번호는 중복되지 않는다.
» 출력
첫째 줄부터 차례대로 N개의 줄에 한줄에 하나씩 시리얼 번호를 정렬한 결과를 출력한다.
예제 입출력(테스트케이스)
입력 | 출력 |
---|---|
5 ABCD 145C A A910 Z321 |
A ABCD Z321 145C A910 |
2 A0A9 A3A0 |
A0A9 A3A0 |
2 3A 11 |
11 3A |
문제 풀이1
fun main(args: Array<String>) = with(System.`in`.bufferedReader()) {
val list = mutableListOf<String>()
repeat(readLine().toInt()) {
list.add(readLine())
}
list.sortedWith(Comparator { o1, o2 ->
if (o1.length != o2.length) o1.length - o2.length
else {
val i1 = intSum(o1)
val i2 = intSum(o2)
if (i1 == i2) o1.compareTo(o2)
else i1.compareTo(i2)
}
}).forEach { println(it) }
}
fun intSum(s: String): Int = s.fold(0) { acc, ch ->
if (ch in '0'..'9') acc + ch.toInt() - 48
else acc
}