[Algorithm] 프로그래머스 - 배열 만들기 6 (Kotlin)

 

java.util.Stack 패키지에 포함된 스택을 써서 문제가 말하는 대로 코드를 작성하면 된다. import문을 써주는 걸 잊지 말자.

 

import java.util.Stack

class Solution {
    fun solution(arr: IntArray): IntArray {
        val stk = Stack<Int>()

        for (element in arr) {
            if (stk.isNotEmpty() && stk.peek() == element) {
                stk.pop()
            } else {
                stk.push(element)
            }
        }

        return if (stk.isEmpty()) {
            intArrayOf(-1)
        } else {
            stk.toIntArray()
        }
    }
}