K = int(input())
stack = []
for _ in range(K):
num = int(input())
if num == 0:
stack.pop()
else:
stack.append(num)
print(sum(stack))
생각보다 간단한 문제였다. num으로 들어온 정수가 0일 때, pop()
함수만 활용하면 쉽게 스택(FILO 구조)에서 Last-In 원소가 pop 된다.
CPython은 시간 초과로 실패하여 PyPy로 제출하였다. input()
대신 sys.stdin.readline()
을 사용하면 통과하지 않을까 싶다.