1181번: 단어 정렬

문제 출처 : https://www.acmicpc.net/problem/1181

1181번: 단어 정렬

언어 : Kotlin

 

문제 설명 :

알파벳 소문자로 이루어진 N개의 단어가 들어오면 아래와 같은 조건에 따라 정렬하는 프로그램을 작성하시오.

1. 길이가 짧은 것부터
2. 길이가 같으면 사전 순으로

단, 중복된 단어는 하나만 남기고 제거해야 한다.

• 첫째 줄에 단어의 개수 N이 주어진다. (1 ≤ N ≤ 20,000) 둘째 줄부터 N개의 줄에 걸쳐 알파벳 소문자로 이루어진 단어가 한 줄에 하나씩 주어진다. 주어지는 문자열의 길이는 50을 넘지 않는다.
• 조건에 따라 정렬하여 단어들을 출력한다.

 

제한 사항 :

• 시간 제한 : 2초
• 메모리 제한 : 256MB

 

입출력 예 :

입력	출력
13 but i wont hesitate no more no more it cannot wait im yours	i im it no but more wait wont yours cannot hesitate

풀이 : 

import java.io.*

fun main(args: Array<String>) {
    val br = BufferedReader(InputStreamReader(System.`in`))
    val bw = BufferedWriter(OutputStreamWriter(System.out))

    val n = br.readLine().toInt()
    var arr = Array(n) { "" }
    repeat(n) {
        arr[it] = br.readLine()
    }

   arr = arr.distinct().sortedWith(compareBy<String> { it.length }.thenComparator { a, b ->
       when {
           a > b -> 1
           a < b -> -1
           else -> 0
       }
   }).toTypedArray()

    arr.forEach { bw.write("${it}\n") }
    bw.flush()
    bw.close()
}