[C++/프로그래머스]코딩테스트 2의 영역

글자지우기

문제 설명

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

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

 

 

제한사항

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

 

 

 

입출력 예

arr	result
[1, 2, 1, 4, 5, 2, 9]	[2, 1, 4, 5, 2]
[1, 2, 1]	[2]
[1, 1, 1]	[-1]
[1, 2, 1, 2, 1, 10, 2, 1]	[2, 1, 2, 1, 10, 2]

 

 

입출력 예 설명

입출력 예 #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 합니다.

 

 

 

C++ 코딩테스트 2의 영역

#include <string>
#include <vector>

using namespace std;

vector<int> solution(vector<int> arr) {
    vector<int> answer;
    int first = -1;
    int last = -1;

    // 배열에서 2가 있는 위치를 모두 찾습니다.
    for (int i = 0; i < arr.size(); i++) {
        if (arr[i] == 2) {
            if (first == -1) {
                first = i;
            }
            last = i;
        }
    }

    // 2가 없는 경우 [-1]을 반환합니다.
    if (first == -1) {
        return {-1};
    }

    // 첫 번째 2의 위치와 마지막 2의 위치 사이의 부분 배열을 반환합니다.
    answer = vector<int>(arr.begin() + first, arr.begin() + last + 1);
    return answer;
}