10773번 : 제로

문제보기

파이썬

import sys

K = int(sys.stdin.readline())
stack = []

for i in range(K):
    n = int(sys.stdin.readline())
    if n == 0:
        stack.pop()
    else:
        stack.append(n)

print(sum(stack))

코드 설명

• 입력받은 수를 stack에 계속 넣어주다가, 0이 들어오면 pop

포인트

stack의 개념을 제대로 이해하고있는지 확인 문제.