[PGS] 배열 조각하기

문제

예제

내 풀이

import java.util.ArrayList;

public class Main {
    public static void main(String[] args) {
        int[] arr = {4,5,4};
        int[] query = {1,0};

        solution(arr, query);
    }

    public static int[] solution(int[] arr, int[] query) {

        // 배열의 요소 담을 ArrayList 선언
        ArrayList<Integer> list = new ArrayList<Integer>();

        // list 에 배열의 요소 담기
        for (int i = 0; i < arr.length; i++) {
            list.add(arr[i]);
        }

        for (int i = 0; i < query.length; i++) {

            // for문 돌 때마다
            ArrayList<Integer> compare = new ArrayList<Integer>();

            // query 값이 짝수
            if(i % 2 == 0){
                // 만약 query 값이 4라면 리스트 0 1 2 3 4에 배열 값을 넣는다
                for(int j = 0; j <= query[i]; j++){
                    compare.add(list.get(j));
                }
            }
            // query 값이 홀수
            else {
                // 만약 query 값이 1이라면 1 부터 arr 의 길이까지 반복문을 돈다
                for (int j = query[i]; j < list.size(); j++) {
                    compare.add(list.get(j));
                }
            }

        list = compare;

        }

        int[] answer = new int[list.size()];
        for(int i = 0; i < list.size(); i++){
            answer[i] = list.get(i);
        }

        return answer;
    }
}