[백준] 10773번: 제로 - Kotlin[코틀린]

문제

 

10773번: 제로

 

---

풀이

 

 Stack을 이용해 풀면 된다. 0을 입력받았을 때 pop()을 호출하고 아닐 때는 push()를 호출해준다. 입력이 끝나면 stack이 비워질 때 까지 pop()을 호출하며 sum 변수에 더해준다.

 

---

코드

 

import java.util.*

fun main() {
    val br = System.`in`.bufferedReader()
    val bw = System.out.bufferedWriter()

    val case = br.readLine().toInt()
    val stack = Stack<Int>()

    repeat(case) {
        val num = br.readLine().toInt()

        if(num == 0) {
            stack.pop()
        } else {
            stack.add(num)
        }
    }

    var sum = 0
    while (stack.isNotEmpty()) {
        sum += stack.pop()
    }

    bw.write("$sum")
    bw.flush()
    bw.close()
    br.close()
}