[프로그래머스] 코딩테스트 연습 - 배열 조각하기(JAVA)

https://school.programmers.co.kr/learn/courses/30/lessons/181893

프로그래머스

 

문제

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

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

• 짝수 인덱스에서는 arr에서 query[i]번 인덱스를 제외하고 배열의 query[i]번 인덱스 뒷부분을 잘라서 버립니다.
• 홀수 인덱스에서는 arr에서 query[i]번 인덱스는 제외하고 배열의 query[i]번 인덱스 앞부분을 잘라서 버립니다.

위 작업을 마친 후 남은 arr의 부분 배열을 return 하는 solution 함수를 완성해 주세요.

---

소스코드

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] ;
            } else {
                start += query[i];
            }
        }

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

 

소스코드 설명

1. 시작 값과 끝 값을 지정

2. 반복문 내부에 조건문을 통해 짝수, 홀수일 때의 계산식 구현

3. Arrays.copyOfRange()를 통해 리턴값 구현