[백준] 1181번: 단어 정렬 - Kotlin[코틀린]

문제

 

1181번: 단어 정렬

 

---

풀이

 

 먼저 코틀린의 HashSet을 이용하여 중복된 단어의 입력을 방지한다. 이 set을 정렬하기 위해 MutableList로 형변환 시켜준다. 이 리스트를 Comparator를 이용하여 정렬시켜주기 위해 sortWith() 함수를 사용한다. 두 단어의 길이가 같은 경우에는 알파벳 순서에 따라 정렬하고, 아닌 경우에는 길이에 따라 정렬해준다. 이 후, 정렬된 리스트를 출력하면 된다. 

 

---

코드

 

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

    val num = br.readLine().toInt()
    val word = hashSetOf<String>()

    repeat(num) {
        word.add(br.readLine())
    }

    word.toMutableList().apply {
        sortWith { d1, d2 ->
            if (d1.length == d2.length) {
                d1.compareTo(d2)
            } else {
                d1.length - d2.length
            }
        }

        this.forEach { // this는 mutableList()이다.
            bw.append(it).append('\n')
        }
    }

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