[Algorithm] 프로그래머스 - 배열 조각하기 (Kotlin)

 

copyOfRange()를 쓰면 비교적 간단하게 풀 수 있다.

 

class Solution {
    fun solution(arr: IntArray, query: IntArray): IntArray {
        var result = arr
        for (i in query.indices) {
            result = if (i % 2 == 0) {
                result.copyOfRange(0, query[i] + 1)
            } else {
                result.copyOfRange(query[i], result.size)
            }
        }
        return result
    }
}