[Algorithm] 프로그래머스 - 접미사인지 확인하기 (Kotlin)

어떤 문자열에 대해서 접미사는 특정 인덱스부터 시작하는 문자열을 의미한다. banana의 모든 접미사는 banana, anana, nana, ana, na, a다. 문자열 my_string과 is_suffix가 주어질 때, is_suffix가 my_string의 접미사면 1, 아니면 0을 리턴하는 함수를 완성하라

 

 

접미사 배열의 풀이를 재활용할 수 있다.

 

class Solution {
    fun solution(my_string: String, is_suffix: String): Int {
        val list: List<String> = my_string.indices
            .map { my_string.substring(it) }
            .sorted()

        return if (list.contains(is_suffix)) 1 else 0
    }
}

 

간결하게 풀면 아래와 같다. endsWith()으로 my_strings가 is_suffix로 끝나는지 확인해서 1 또는 0을 리턴한다.

 

class Solution {
    fun solution(my_string: String, is_suffix: String): Int =
        if (my_string.endsWith(is_suffix)) 1 else 0
}