[Python] 백준 10773 제로

10773번: 제로

스택은 파이썬의 배열로 구현이 가능하다.

 

1) 0이면 스택 팝

2) 0이 아니면 스택에 넣기

3) 스택의 합 출력

 

전체 코드

import sys
input = sys.stdin.readline

K = int(input())
stack = []
for _ in range(K):
    a = int(input())
    if a == 0:
        stack.pop()
    else:
        stack.append(a)

print(sum(stack))