[백준 10773번 문제 파이썬(python)] [제로]

문제

설명

스택을 만들어서 0이 아닐때에는 값을 넣고 0일때는 가장 최근 스택값을 없애주고 마지막에 합을 구해주면 되는 문제

 

코드

import sys

K = int(sys.stdin.readline())
stk = []
for _ in range(K):
    number = int(sys.stdin.readline())
    if number != 0:
        stk.append(number)
    else:
        stk.pop()

print(sum(stk))