[프로그래머스] 2의 영역

 

 

• 풀이

#include <string>
#include <vector>

using namespace std;

vector<int> solution(vector<int> arr) {
    vector<int> answer;
    int first=-1, last=1;
    for(int i=0; i<arr.size(); i++){
        if(arr[i] == 2){
            if(first == -1) first = i;
            last = i;
        }
    }
    if(first==-1) return {-1};
    answer = vector<int>(arr.begin()+ first, arr.begin()+last +1);
    return answer;
}