[프로그래머스/Java] Lv.0 배열 만들기 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의 길이

풀이

import java.util.*;

class Solution {
    public int[] solution(int[] arr, int[][] intervals) {
        List<Integer> temp = new ArrayList<>();
        int[] newArr = {};
        
        for(int[] interval : intervals){
            int a = interval[0];
            int b = interval[1];
            
            newArr = Arrays.copyOfRange(arr, a, b+1);
            
            for(int i=0;i<newArr.length;i++){
                temp.add(newArr[i]);
            }
        }
        
        int[] answer = new int[temp.size()];
        for(int i=0;i<temp.size();i++){
            answer[i] = temp.get(i);
        }
        
        return answer;
    }
}

결과