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

문제

공백으로 구분하기 1 : 문제 링크

문제 분석

• 단어가 공백 한 개로 구분되어 있는 문자열 my_string이 매개변수로 주어질 때, my_string에 나온 단어를 앞에서부터 순서대로 담은 문자열 배열을 return
• for loop에서 if문을 통해 my_string에서 공백을 만날때까지 temp라는 문자열 변수에 단어를 저장. 공백을 만날경우 else문을 통해 temp에 저장된 단어를 answer에 저장하고, temp는 다시 초기화. for loop가 끝난후 마지막에 저장된 단어를 answer를 저장하고, 단어가 하나일 경우를 대비하여 push_back을 한번 더 수행. 최종적으로 저장된 answer 벡터를 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.size(); i++) {
        if(my_string[i] != ' ') temp += my_string[i];
        else {
            answer.push_back(temp);
            temp = "";
        }
    }
    answer.push_back(temp);
    return answer;
}