Your Name

I am an amazing person.

1 분 소요

문제 설명

프로그래머스 181862번 세 개의 구분자

임의의 문자열이 주어졌을 때 문자 “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”]

입출력 예 설명 1

  • 문제 설명의 예시와 같습니다.

입출력 예 설명 2

  • “c” 이전에는 “a”, “b”, “c” 이외의 문자가 없습니다.
  • “c” 이후에 문자열 “d”가 있으므로 “d”를 저장합니다.
  • 따라서 [“d”]를 return 합니다.

입출력 예 설명 3

  • “a”, “b”, “c” 이외의 문자가 존재하지 않습니다. 따라서 저장할 문자열이 없습니다.
  • 따라서 [“EMPTY”]를 return 합니다.

코드 구현

Python

def solution(myStr: str):
    a_split = myStr.split("a")
    b_split = []
    answer = []

    for item in a_split:
        splits = item.split("b")
        for split in splits:
            if split != "":
                b_split.append(split)

    for item in b_split:
        splits = item.split("c")
        for split in splits:
            if split != "":
                answer.append(split)

    return ["EMPTY"] if len(answer) == 0 else answer

Java

import java.util.ArrayList;
import java.util.List;

class Solution {
    public String[] solution(String myStr) {
        ArrayList<String> splitA = new ArrayList<>(List.of(myStr.split("a")));
        ArrayList<String> splitB = new ArrayList<>();
        ArrayList<String> answer = new ArrayList<>();

        for (String s : splitA) {
            String[] split = s.split("b");
            for (String s2 : split) {
                if (!s2.equals("")) {
                    splitB.add(s2);
                }
            }
        }

        for (String s : splitB) {
            String[] split = s.split("c");
            for (String s2 : split) {
                if (!s2.equals("")) {
                    answer.add(s2);
                }
            }
        }

        return answer.isEmpty() ? new String[] {"EMPTY"} : answer.toArray(new String[0]);
    }
}

참고