[백준] 1181번 : 단어 정렬 (Kotlin)

1181번: 단어 정렬

나의 풀이

fun main(args: Array<String>) {
    val n = readLine()!!.toInt()
    val arr = ArrayList<String>()

    repeat(n) {
        val input = readLine()!!.toString()
        arr.add(input)
    }
    // distinct()는 List 반환
    val arr2 = arr.distinct().toTypedArray()

    val result= arr2.sortedWith(Comparator { o1, o2 ->
        when {
            o1.length > o2.length -> { 1 }
            o1.length < o2.length -> { -1 }
            else -> {
                o1.compareTo(o2)
            }
        }
    })
    result.forEach {
        println(it)
    }
}

☞ 중복을 제거한 Array를 준비합니다. 그리고 sortedWith(Comparator{ ... }) 구문을 통해 정렬합니다.

문자열의 length로 먼저 정렬하고, length가 같은 경우 사전순으로 정렬합니다.