배열 만들기 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> list = new ArrayList<>();
        for(int i=intervals[0][0]; i<=intervals[0][1]; i++) {
            list.add(arr[i]);
        }
        for(int j=intervals[1][0]; j<=intervals[1][1]; j++) {
            list.add(arr[j]);
        }
        int[] answer = list.stream().mapToInt(i->i).toArray();
        return answer;
    }
}

다른 사람 코드

class Solution {
    public int[] solution(int[] arr, int[][] intervals) {

        int s1 = intervals[0][0], e1 = intervals[0][1], s2 = intervals[1][0], e2 = intervals[1][1], idx = 0;
        int[] answer = new int[e1 - s1 + e2 - s2 + 2];

        for (int i = s1; i <= e1; i++) {
            answer[idx++] = arr[i];
        }

        for (int i = s2; i <= e2; i++) {
            answer[idx++] = arr[i];
        }

        return answer;
    }
}

느낀 점

처음 생각한 건 다른 사람 코드에 나와있는 것처럼 따로 List로 변환하는 과정을 거치지 않고, answer배열에 범위를 정하고 반복문으로 채워넣는 것이었다. 그러나 int 변수 idx를 두고 따로 ++시키는 방법을 생각해내지 못해 List를 사용하여 문제를 해결했다. 이전 문제에서 활용했던 list.stream().mapToInt().toArray() 메소드를 활용하였다 :)