[백준/Python] 10773 제로

[백준/Python] 10773 제로

풀이 코드 및 설명

import sys

input = sys.stdin.readline

K = int(input())
stack = []
for _ in range(K):
    n = int(input())
    if n > 0: 
        stack.append(n)
    elif n == 0:
        stack.pop()
sum = 0
for i in stack:
    sum += i

print(sum)    

1. 스택을 사용해서 0 이 아닌 수는 스택에 삽입
2. 0 이라면 pop() 을 통해서 제일 최근에 삽입된 값을 제거 한다.
3. 스택 내 모든 원소의 합을 출력한다.