프로그래머스 - 2의 영역 - C++

https://school.programmers.co.kr/learn/courses/30/lessons/181894

프로그래머스

 

#include <string>
#include <vector>

using namespace std;

vector<int> solution(vector<int> arr) {
    vector<int> answer
    for(int i=0;i<arr.size();i++)
    {
        if(arr[i]==2){
            si=i;break;
        }
    }
    if(si==-1){
        answer.push_back(-1);return answer;
    }
    for(int i=arr.size()-1;i>=0;i--)
    {
        if(arr[i]==2)
        {
            se=i;break;
        }
    }
    for(int i=si;i<=se;i++)answer.push_back(arr[i]);
    return answer;
}

 

설명

첫번째 2의 위치부터 마지막 2까지의 배열을 리턴해주면 되는 문제이다. 

따라서 0 번 인덱스부터 첫번째 2까지의 위치를 찾고, 맨 뒤부터 제일 뒤에 있는 2가 나올 때까지의 위치를 찾는다.