[Algorithm] 프로그래머스 - 접미사 배열 (Kotlin)

어떤 문자열에 대해 접미사는 특정 인덱스부터 시작하는 문자열을 의미한다. banana의 모든 접미사는 banana, anana, nana, ana, na, a다. 문자열 my_string이 매개변수로 주어질 때 이 문자열의 모든 접미사를 사전순으로 정렬한 문자열 배열을 리턴하는 함수를 완성하라

 

 

첫 글자부터 하나씩 사라지는 형태기 때문에 0부터 문자열의 전체 길이까지의 range를 만든 다음 substring()과 sorted()를 써서 사전 순으로 정렬한다. 이후 리턴타입에 맞춰 배열로 바꿔서 리턴하면 된다. sorted()의 리턴 타입이 List<T>기 때문이다.

 

class Solution {
    fun solution(my_string: String): Array<String> =
        (0 until my_string.length)
            .map { my_string.substring(it) }
            .sorted()
            .toTypedArray()
}