[프로그래머스 / C++] 왼쪽 오른쪽

문제

왼쪽 오른쪽 : 문제 링크

문제 분석

• 문자열 리스트 str_list에는 "u","d","l","r" 네 개의 문자열이 여러 개 저장되어 있다. str_list에서 "l"과 "r" 중 먼저 나오는 문자열이 "l"이라면 해당 문자열을 기준으로 왼쪽에 있는 문자열들을 순서대로 담은 리스트를, 먼저 나오는 문자열이 "r"이라면 해당 문자열을 기준으로 오른쪽에 있는 문자열들을 순서대로 담은 리스트를 return ("l"이나 "r"이 없다면 빈 리스트를 return)
• "l"이나 "r"이 없을 경우를 대비하여 빈 문자열 벡터 answer과 "l" 또는 "r"의 인덱스를 저장할 정수형 변수 pos를 0으로 초기화. for loop를 통해 str_list의 첫번째부터 마지막 원소까지 순환. if문을 통해 "l"을 발견했을 경우, pos에 현재 인덱스를 저장하고 str_list의 첫번째부터 해당 인덱스의 이전 위치까지의 문자열 벡터를 return. else if문을 통해 "r"을 발견했을 경우, pos에 현재 인덱스를 저장하고 해당 인덱스의 다음 위치부터 마지막까지의 문자열 벡터를 return. loop 탈출 후, pos가 0이라면 "l"이나 "r"이 없음을 의미하므로 빈 리스트 answer을 return

풀이

#include <string>
#include <vector>

using namespace std;

vector<string> solution(vector<string> str_list) {
    vector<string> answer;
    int pos = 0;
    
    for(int i = 0; i < str_list.size(); ++i) {
        if(str_list[i] == "l") {
            pos = i;
            return vector<string>(str_list.begin(), str_list.begin() + pos);
        }
        else if(str_list[i] == "r") {
            pos = i;
            return vector<string>(str_list.begin() + pos + 1, str_list.end());
        }
    }
    if(pos == 0) return answer;
}