백준 알고리즘 10773번 제로(python)

기본적인 스택 문제이다.

입력이 0이 아니면 스택에 추가하고 입력이 0이면 제일 마지막에 추가한 수를 스택에서 제거한다.

마지막에 스택에 들어있는 수들의 합을 출력해주면 된다.

import sys 

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