제로(백준)

package com.example.algorithm

import java.io.BufferedReader
import java.io.BufferedWriter
import java.io.InputStreamReader
import java.io.OutputStreamWriter
import java.util.*


fun main(){

    val br = BufferedReader(InputStreamReader(System.`in`))
    val bw = BufferedWriter(OutputStreamWriter(System.`out`))

    val count = br.readLine().toInt()

    val stack = Stack<Int>()
    repeat(count){
        val input = br.readLine().toInt()
        if(input == 0){
            stack.pop()
        }else{
            stack.add(input)
        }
    }

    bw.write(String.format("%d",stack.sum()))

    bw.flush()
    bw.close()
    br.close()
}

문제를 딱 보고 스택을 사용해서 풀면되겠다고 생각했다.

잘못 입력할 시 가장 최근에 쓴 숫자를 지우면 되므로 

input이 0인지 판별하여 스택으로 구현해주면 된다.