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

문제설명

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

제한사항

• my_string은 영소문자와 공백으로만 이루어져 있습니다.
• 1 ≤ my_string의 길이 ≤ 1,000
• my_string의 맨 앞과 맨 뒤에 글자는 공백이 아닙니다.

입출력 예

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

풀이

my_string 문자열에서 공백을 기준으로 문자열을 구분하여 문자열 배열을 return하는 구현 문제입니다.

#include <string>
#include <vector>

using namespace std;

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

결과