알고리즘/백준

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

알감자 2021. 12. 27. 22:27
 

10773번: 제로

첫 번째 줄에 정수 K가 주어진다. (1 ≤ K ≤ 100,000) 이후 K개의 줄에 정수가 1개씩 주어진다. 정수는 0에서 1,000,000 사이의 값을 가지며, 정수가 "0" 일 경우에는 가장 최근에 쓴 수를 지우고, 아닐 경

www.acmicpc.net

 

코드

stack = []
def count(i):
    if i == 0:
        if len(stack) >= 1:
            stack.pop(-1)
    else:
        stack.append(i)
    return

k = int(input())
num_list = list(int(input()) for i in range(k))
for i in num_list:
    count(i)

print(sum(stack))