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

2의 영역

---

문제설명

정수 배열 arr가 주어집니다. 배열 안의 2가 모두 포함된 가장 작은 연속된 부분 배열을 return 하는 solution 함수를 완성해 주세요.

단, arr에 2가 없는 경우 [-1]을 return 합니다.

---

제한사항

• 1 ≤ arr의 길이 ≤ 100,000
  • 1 ≤ arr의 원소 ≤ 10

---

입출력 예

입출력 예 설명

입출력 예 #1

• 2가 있는 인덱스는 1번, 5번 인덱스뿐이므로 1번부터 5번 인덱스까지의 부분 배열인 [2, 1, 4, 5, 2]를 return 합니다.

입출력 예 #2

• 2가 한 개뿐이므로 [2]를 return 합니다.

입출력 예 #3

• 2가 배열에 없으므로 [-1]을 return 합니다.

입출력 예 #4

• 2가 있는 인덱스는 1번, 3번, 6번 인덱스이므로 1번부터 6번 인덱스까지의 부분 배열인 [2, 1, 2, 1, 10, 2]를 return 합니다.

---

작성코드

#include <string>
#include <vector>

using namespace std;

vector<int> solution(vector<int> arr) {
    vector<int> answer;
    int first = -1;
    int 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;
}

---

문제 URL

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

프로그래머스

---