Kotlin 🍬 백준 13단계 :: 1181 번

Hello World

문제   |

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

  1. 길이가 짧은 것부터

  2. 길이가 같으면 사전 순으로

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

 

 

문제   |

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

 

문제   |

  조건에 따라 정렬하여 단어들을 출력한다.

 

 

풀이  |

  sortedWith()와 인자로 compareBy()를 사용한다.

 

 

답안  |

import java.util.Scanner

fun main() = with(Scanner(System.`in`)) {
    val N = nextInt()
    val arr = mutableSetOf<String>()

    repeat(N) {
        arr.add(next())
    }
    
    val result = arr.sortedWith(compareBy({ it.length }, { it }))
    result.forEach { word ->
        println(word)
    }
}