[백준/Python] 10773. 제로

 

import sys
input = sys.stdin.readline # 시간초과를 방지하기 위한 코드

N = int(input())
stack = []
for i in range(N):
    tmp = int(input())
    if tmp == 0: # 0이 입력된다면 stack의 마지막부분을 제거
        stack.pop()
    else:
        stack.append(tmp)

print(sum(stack)) # for문이 끝나면 남아있는 stack의 합을 구함