(Python) 백준 10773번 - 제로

문제 링크 : https://www.acmicpc.net/problem/10773

10773번: 제로

풀이

스택을 사용해 0이 아닌 수는 저장하고, 0인 경우에는 pop()하는 것으로 풀이를 진행했다.

import sys
from collections import deque

arr = deque()
for _ in range(int(sys.stdin.readline())):
    n = int(sys.stdin.readline())
    if n == 0: # 0이면 가장 최근의 수 삭제
        arr.pop()
    else: # 0이 아니면 arr에 추가
        arr.append(n) 
sum = 0
for i in arr:
    sum += i # arr 내의 모든 수의 합 계산
print(sum)