알고리즘/프로그래머스 1단계

프로그래머스 - 왼쪽 오른쪽 - C++

게임만드는학생 2023. 7. 31. 11:08

https://school.programmers.co.kr/learn/courses/30/lessons/181890?language=cpp 

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

#include <string>
#include <vector>

using namespace std;

vector<string> solution(vector<string> s) {
    vector<string> answer;

    for (int i = 0; i < s.size(); i++)
    {
        if (s[i] == "l" || s[i] == "r")
        {
            if (s[i] == "l")
            {
                if (i == 0)return answer;
                for (int j = 0; j < i; j++)
                {
                    answer.push_back(s[j]);
                }break;
            }
            else
            {
                if (i == s.size() - 1)return answer;
                for (int j = i + 1; j < s.size(); j++)answer.push_back(s[j]);
                break;
            }
        }
    }
    return answer;
}

 

설명

for문을 돌며 가장 먼저 나오는 l 또는 r 을 찾아서 해당 인덱스를 기준으로 모든 문자를 answer에 추가하면된다.

여기서 예외처리로 l 이 0 번째이거나 r이 마지막인 경우를 처리해주면 된다.