백준(BAEKJOON) 10773 제로 - 파이썬(python)

 

 

스택으로 접근해서 풀이

0인 경우 삭제할 수 있는 데이터가 무조건 존재한다는 가정이 있기 때문에 별도의 조건 필요 없음.

 

import sys

iter_num = int(sys.stdin.readline())
budget = []

for i in range(iter_num):
    x = int(sys.stdin.readline())
    if x == 0:
        budget.pop()
    else:
        budget.append(x)

print(sum(budget))