[백준 10773] 제로 with Python

문제 링크

www.acmicpc.net/problem/10773

10773번: 제로

풀이

import sys

k = int(sys.stdin.readline())
arr = []
for i in range(k):
    n = int(sys.stdin.readline())
    if n == 0:
        arr.pop()
    else:
        arr.append(n)
print(sum(arr))

0을 입력받으면 가장 마지막 요소를 제거하고(pop) 나머지는 맨 마지막에 추가(push)하는 스택을 구현하여 해결하였다.