배열 만들기 3

문제 설명

정수 배열 arr와 2개의 구간이 담긴 배열 intervals가 주어집니다.

intervals는 항상 [[a1, b1], [a2, b2]]의 꼴로 주어지며 각 구간은 닫힌 구간입니다. 닫힌 구간은 양 끝값과 그 사이의 값을 모두 포함하는 구간을 의미합니다.

이때 배열 arr의 첫 번째 구간에 해당하는 배열과 두 번째 구간에 해당하는 배열을 앞뒤로 붙여 새로운 배열을 만들어 return 하는 solution 함수를 완성해 주세요.

 

제한사항

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

 

입출력 예

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

---

import java.util.ArrayList;
import java.util.Arrays;

class Solution {
    public ArrayList<Integer> solution(int[] arr, int[][] intervals) {
        ArrayList<Integer> answer = new ArrayList<Integer>();
        
        for(int i=0; i<intervals.length; i++) {
                // 예를 들어 1, 3이 리턴된다면 arr에서 index 1~3을 뽑아야 한다.
            int[] newarr = Arrays.copyOfRange(arr, intervals[i][0], intervals[i][1]+1);
            for(int e : newarr) {
                answer.add(e);
            }
        }
        return answer;
    }
}

---

이번 코드는 soso한 거 같다.

 

다른 사람의 코드를 보니

import java.util.*;

 

class Solution {

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

        return Arrays.stream(intervals).flatMapToInt(ints -> Arrays.stream(Arrays.copyOfRange(arr, ints[0], ints[1] + 1))).toArray();

    }

}

이렇게 했더라.

 

내가 자바 내에서의 자료구조, 메서드를 거의 모르는 것 같다.

 

*문제에서 int[] 반환하라고 했는데 내가 ArrayList<Integer> 이런 식으로 메서드의 시그니처를 바꿨다.

절대 하면 안 되는 행동 같다.

그래도 이번에 이렇게 해 보면서 하면 안 되는 행위란 걸 알았다.