[Lv. 0] 세 개의 구분자

문제 출처 : https://school.programmers.co.kr/learn/courses/30/lessons/181862

프로그래머스

난이도 : Level.0

언어 : Kotlin

 

문제 설명 :

임의의 문자열이 주어졌을 때 문자 "a", "b", "c"를 구분자로 사용해 문자열을 나누고자 합니다.

 

예를 들어 주어진 문자열이 "baconlettucetomato"라면 나눠진 문자열 목록은 ["onlettu", "etom", "to"] 가 됩니다.

 

문자열 myStr이 주어졌을 때 위 예시와 같이 "a", "b", "c"를 사용해 나눠진 문자열을 순서대로 저장한 배열을 return 하는 solution 함수를 완성해 주세요.

 

단, 두 구분자 사이에 다른 문자가 없을 경우에는 아무것도 저장하지 않으며, return할 배열이 빈 배열이라면 ["EMPTY"]를 return 합니다.

 

제한 사항 : 

• 1 ≤ myStr의 길이 ≤ 1,000,000
  • myStr은 알파벳 소문자로 이루어진 문자열 입니다.

입출력 예 :

myStr	result
"baconlettucetomato"	["onlettu", "etom", "to"]
"abcd"	["d"]
"cabab"	["EMPTY"]

풀이 : 

class Solution {
    fun solution(myStr: String): Array<String> {
        var answer = myStr.map { if (it == 'a' || it == 'b' || it == 'c') '0' else it }.joinToString("").split("0").filter { it.isNotEmpty() }.toTypedArray() 
        
        return if (answer.isEmpty()) {
            arrayOf("EMPTY")
        } else {
            answer
        }
    }
}