[Algorithm] 프로그래머스 - 빈 배열에 추가, 삭제하기 (Kotlin)

 

true면 해당 원소에 2를 곱한 만큼 배열에 추가하고, false면 그 원소만큼 배열에서 제거한다.

forEachIndexed를 쓰면 간단하게 풀 수 있다.

 

class Solution {
    fun solution(arr: IntArray, flag: BooleanArray): IntArray {
        val answer = arrayListOf<Int>()
        flag.forEachIndexed { i, b ->
            if (b) {
                repeat(arr[i] * 2) {
                    answer.add(arr[i])
                }
            } else {
                repeat(arr[i]) {
                    answer.removeLast()
                }
            }
        }

        return answer.toIntArray()
    }
}