[C++] 프로그래머스 공백으로 구분하기 2

문제설명

단어가 공백 한 개 이상으로 구분되어 있는
문자열 my_string이 매개변수로 주어질 때,
my_string에 나온 단어를 앞에서부터 순서대로 담은 문자열 배열을
return 하는 solution 함수를 작성해 주세요.

제한사항

• my_string은 영소문자와 공백으로만 이루어져 있습니다.
• 1 ≤ my_string의 길이 ≤ 1,000
• my_string의 맨 앞과 맨 뒤에도 공백이 있을 수 있습니다.
• my_string에는 단어가 하나 이상 존재합니다.

입출력 예

my_string	result
” i love you”	[“i”, “love”, “you”]
” programmers “	[“programmers”]

풀이

my_string 문자열에서 공백을 기준으로 문자열을 구분하여 문자열 배열을 return하는 구현 문제입니다. 공백이 한 개 이상으로 구분되어 있어서 문자열 변수를 하나 선언하여 공백이 아닐 경우에만 answer 배열에 문자열을 넣도록 구현하여 문제를 풀 수 있었습니다.

#include <string>
#include <vector>

using namespace std;

vector<string> solution(string my_string) {
    vector<string> answer;
    string tmp = "";
    for(int i = 0; i < my_string.length(); i++)
    {
        if(my_string[i] == ' ')
        {
            if(tmp != "")
            {
                answer.push_back(tmp);
                tmp = "";
            }
        }
        else
        {
            tmp.push_back(my_string[i]);
        }
    }
    
    if(tmp != "")
        answer.push_back(tmp);
    
    return answer;
}

결과