[C++] 프로그래머스 배열 만들기 3

문제설명

정수 배열 arr와 2개의 구간이 담긴 배열 intervals가 주어집니다.
intervals는 항상 [[a1, b1], [a2, b2]]의 꼴로 주어지며 각 구간은 닫힌 구간입니다.
닫힌 구간은 양 끝값과 그 사이의 값을 모두 포함하는 구간을 의미합니다.
이때 배열 arr의 첫 번째 구간에 해당하는 배열과 두 번째 구간에 해당하는
배열을 앞뒤로 붙여 새로운 배열을 만들어 return 하는 solution 함수를 완성해 주세요.

제한사항

• 1 ≤ arr의 길이 ≤ 100,000
  ◦ 1 ≤ arr의 원소 < 100
• 1 ≤ a1 ≤ b1 < arr의 길이
• 1 ≤ a2 ≤ b2 < arr의 길이

입출력 예

arr	num_list	result
[1, 2, 3, 4, 5]	[[1, 3], [0, 4]]	[2, 3, 4, 1, 2, 3, 4, 5]

풀이

intervals에 해당하는 배열을 추출하고 이어붙이는 간단한 구현 문제입니다.

#include <string>
#include <vector>

using namespace std;

vector<int> solution(vector<int> arr, vector<vector<int>> intervals) {
    vector<int> answer;
    
    for(int i = 0; i < 2; i++)
    {
        for(int j = intervals[i][0]; j <= intervals[i][1]; j++)
        {
            answer.push_back(arr[j]);
        }
    }
    
    return answer;
}

결과