[백준] 10773번: 제로 - 코틀린(kotlin)

https://www.acmicpc.net/problem/10773

10773번: 제로

 

문제 해석

0이 주어지면 가장 최근에 쓴 수를 지웁니다.

또한 0이 주어지면 항상 지울 수 있는 수가 존재한다는 조건이 존재합니다.

즉, 스택을 활용해서 0이 들어오는 순간에는 POP을 하면 될 것 같습니다.

 

코드

fun main() {
    val stack = ArrayDeque<Int>()
    val count = readLine()!!.toInt()
    repeat(count){
        val curNum = readLine()!!.toInt()
        if(curNum == 0){
            stack.removeLast()
        }
        if(curNum != 0 ) {
            stack.addLast(curNum)
        }
    }
    println(stack.sum())
}

비권장되는 stack대신 ArrayDeque를 활용하여 해결했습니다.