[백준/파이썬] 10773번

백준 10773번

문제

https://www.acmicpc.net/problem/10773

코드

import sys

n = int(sys.stdin.readline())

result = []
sum = 0

for _ in range(n):
    num = int(sys.stdin.readline())
    if num == 0:
        result.pop()
    else:
        result.append(num)

for i in result:
    sum += i

print(sum)

풀이

스택을 이용해서 풀면됩니다.
입력받은 값이 0인 경우, result에 저장해준 마지막 값 (top) 값을 빼주면 됩니다. (pop)
0이 아니면, result에 더해줍니다. (push)