[JAVA] 배열 조각하기 - Arrays.copyOfRange()

정수 배열 arr와 query가 주어집니다.

query를 순회하면서 다음 작업을 반복합니다.

짝수 인덱스에서는 arr에서 query[i]번 인덱스를 제외하고 배열의 query[i]번 인덱스 뒷부분을 잘라서 버립니다.
홀수 인덱스에서는 arr에서 query[i]번 인덱스는 제외하고 배열의 query[i]번 인덱스 앞부분을 잘라서 버립니다.
위 작업을 마친 후 남은 arr의 부분 배열을 return 하는 solution 함수를 완성해 주세요.

 

풀이

import java.util.*;
class Solution {
    public List<Integer> solution(int[] arr, int[] query) {
        List<Integer> answer = new ArrayList<>();
        for(int i=0; i<arr.length; i++){
            answer.add(arr[i]);
        }
        
        for(int i=0; i<query.length; i++){
            if(i%2==0){
                int len = answer.size();
                for(int j=query[i]+1;j<len;j++){
                    answer.remove(query[i]+1);
                }
            } else {
                for(int j=0;j<query[i];j++){
                    answer.remove(0);
                }
            }
        }
        return answer;
    }
}

---

import java.util.*;

class Solution {
    public int[] solution(int[] arr, int[] query) {
        int start = 0;
        int end = arr.length - 1;
        for (int i = 0; i < query.length; i++) {
            if (i % 2 == 0) {
                end = start + query[i] - 1;
            } else {
                start += query[i];
            }
        }

        return Arrays.copyOfRange(arr, start, end + 2);
    }
}

 

Arrays.copyOfRange(복사할배열, 시작, 끝배열);