[백준] 1431번: 시리얼 번호 - Kotlin[코틀린]

문제

 

1431번: 시리얼 번호

 

---

풀이

 

 시리얼 넘버와 자리수의 합을 포함하는 Guitar 데이터 클래스를 정의하고 Comparable 인터페이스를 구현하여 풀이하였다. 

 

---

코드

 

data class Guitar(val serialNumber: String, val digitalSum: Int): Comparable<Guitar> {
    override fun compareTo(other: Guitar): Int {
        return if(serialNumber.length != other.serialNumber.length) {
            serialNumber.length - other.serialNumber.length
        } else {
            if(digitalSum != other.digitalSum) {
                digitalSum - other.digitalSum
            } else {
                serialNumber.compareTo(other.serialNumber)
            }
        }
    }
}

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

    val list = mutableListOf<Guitar>()

    repeat(br.readLine().toInt()) {
        val serial = br.readLine()
        var sum = 0 // 자리수의 합
        serial.forEach {
            if(it.isDigit()) { // 숫자일 경우
                sum += it - '0'
            }
        }

        list.add(Guitar(serial, sum))
    }

    list.sort()
    list.forEach {
        bw.append("${it.serialNumber}\n")
    }

    bw.write("")
    bw.flush()
    bw.close()
    br.close()
}