[C++] 프로그래머스 세 개의 구분자

문제설명

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

풀이

문자 a, b, c를 구분자로 이용하여 문자열을 나누는 문제입니다.

#include <string>
#include <vector>

using namespace std;

vector<string> solution(string myStr) {
    vector<string> answer;
    string temp;
    
    for(int i = 0; i < myStr.length(); i++)
    {
        if(myStr[i] != 'a' && myStr[i] != 'b' && myStr[i] != 'c')
        {
            temp.push_back(myStr[i]);
        }
        else
        {
            if(!temp.empty())
            {
                answer.push_back(temp);
                temp = "";
            }
        }
    }
    
    if(!temp.empty())
        answer.push_back(temp);
    
    if(answer.size() == 0)
        answer.push_back("EMPTY");
    
    return answer;
}

결과