[프로그래머스 코딩테스트] 배열 만들기 3(Java)

  문제 설명

  정수 배열 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	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 int[] solution(int[] arr, int[][] intervals) {
        ArrayList<Integer> list = new ArrayList<>();
        for (int i = 0; i < intervals.length; i++) {
            int[] array = Arrays.copyOfRange(arr, intervals[i][0], intervals[i][1] + 1);
            for (int j = 0; j < array.length; j++) {
                list.add(array[j]);
            }
        }
        int[] answer = new int[list.size()];
        int index = 0;
        for (int i = 0; i < list.size(); i++) {
            answer[index++] = list.get(i);
        }
        return answer;
    }
}

  ArrayList를 생성합니다. 정수형이 들어갈 것이므로 제네릭 타입으로는 Integer를 지정합니다.

  바깥 for문이 intervals의 길이만큼 순회합니다.

  int 배열을 생성한 뒤 intervals에서 자르고자 하는 배열의 길이만큼 인덱싱하여 배열을 복사한 값을 초기화 값으로 지정합니다. + 1을 하는 이유는 copyOfRange 메소드가 종료 지점의 값은 포함하지 않기 때문에, 종료 지점의 값까지 포함하기 위해 더해준 것입니다.

  안쪽 for문이 arr의 길이만큼 순회합니다. list에 array의 j번째 원소를 추가합니다. Array와 ArrayList간의 복사는 메소드로 할 수 없으므로 arr의 j번째 원소를 list에 추가함으로써 간접적으로 복사했습니다.

  answer 배열을 생성합니다. list를 복사한 배열이 될 것이므로 크기 역시  list와 동일하게 맞춰줍니다.

  index 변수를 생성합니다. 이는 answer에 원소를 삽입할 때, answer의 인덱스가 될 것입니다.

  for문이 list의 크기만큼 순회합니다. answer의 index번째 원소에 list의 i번째 원소를 할당합니다. 앞서 복사한 것과 마찬가지로, 메소드를 이용하여 ArrayList를 Array에 복사할 수 없으므로 이런 과정을 거치는 것입니다.