백준 10773 - 제로 python

스택을 활용하는 기본 문제이다.

 

import sys

input = sys.stdin.readline

K = int(input())

stack = []
for i in range(K):
    num = int(input())

    if num == 0:
        stack.pop(-1)
    else:
        stack.append(num)
print(sum(stack))

 

---

 

www.acmicpc.net/problem/10773

10773번: 제로