Lv.0 세 개의 구분자

문제 설명

임의의 문자열이 주어졌을 때 문자 "a", "b", "c"를 구분자로 사용해 문자열을 나누고자 합니다.
예를 들어 주어진 문자열이 "baconlettucetomato"라면 나눠진 문자열 목록은 ["onlettu", "etom", "to"] 가 됩니다.
문자열 myStr이 주어졌을 때 위 예시와 같이 "a", "b", "c"를 사용해 나눠진 문자열을 순서대로 저장한 배열을 return 하는 solution 함수를 완성해 주세요.
단, 두 구분자 사이에 다른 문자가 없을 경우에는 아무것도 저장하지 않으며, return할 배열이 빈 배열이라면 ["EMPTY"]를 return 합니다.

입출력

코드 풀이

def solution(myStr):
    # 결과를 저장할 리스트 초기화
    result = []
    
    # 문자열을 순회하면서 구분자를 기준으로 문자열 나누기
    current_str = ""
    for char in myStr:
        if char in ["a", "b", "c"]:
            if current_str:
                result.append(current_str)
            current_str = ""
        else:
            current_str += char
    
    # 마지막으로 남은 문자열 추가
    if current_str:
        result.append(current_str)
    
    # 결과 리스트가 비어있는 경우 "EMPTY" 반환
    return result if result else ["EMPTY"]

출처: 프로그래머스 코딩테스트 연습