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

문제

배열 만들기 3 : 문제 링크

문제 분석

• 정수 배열 arr와 2개의 구간이 담긴 배열 intervals가 주어진다. intervals는 항상 [[a1, b1], [a2, b2]] 의 꼴로 주어지며 각 구간은 닫힌 구간이다. 닫힌 구간은 양 끝 값과 그 사이의 값을 모두 포함하는 구간을 의미한다. 이때 배열 arr의 첫 번째 구간에 해당하는 배열과 두 번째 구간에 해당하는 배열을 앞뒤로 붙여 새로운 배열을 만들어 return
• 이중 for loop를 사용하여 첫번째 loop는 2개의 구간을 검사하고, 두번째 for loop는 각 구간 사이의 값([a, b])을 모두 정수형 배열에 저장. 최종적으로 저장된 answer를 return

풀이

#include <vector>

using namespace std;

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