배열 조각하기

문제 설명

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

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

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

제한사항

5 ≤ arr의 길이 ≤ 100,000
0 ≤ arr의 원소 ≤ 100
1 ≤ query의 길이 < min(50, arr의 길이 / 2)
query의 각 원소는 0보다 크거나 같고 남아있는 arr의 길이 보다 작습니다.

나의 코드

• 첫 번째 코드

import java.util.*;

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

코드 실행은 통과하였으나 제출 후 채점하기에서 실패

• 최종 코드

import java.util.Arrays;

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

Arrays.copyOfRange() 메소드 사용

느낀 점

첫 번째 코드에선 문제를 잘 이해하지 못했다.. 너무 입출력 예에만 몰입해서 짝수면 맨 뒤 요소만 잘라버리고, 홀수면 맨 앞 요소만 잘라버리는 코드를 작성했다. 정신차리고 초기화 후에 Arrays.copyOfRange() 메소드를 사용하여 새로운 방식으로 접근했다. query[i]를 기준으로 잘라냈고, 이상없이 통과! 문제를 잘 읽자 🥲