152. 10773(제로) - 파이썬

import sys
input = sys.stdin.readline

K = int(input())
stack = []
result = 0

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

print(sum(stack))

문제요약

재민이가 잘못 된 수를 외치면 0을 외쳐서 지우게 함. 

재민이가 적은 수의 합을 알아보자

설명

아주 간단한 문제이다.

0이 들어오면 스택에서 꺼내면 된다.

 

  if N == 0 :
    stack.pop()
  else :
    stack.append(N)

만약 0이 들어오면 pop으로 마지막 수를 지우고 그게 아니면 값을 집어 넣는다.

 

print(sum(stack))

sum 함수를 통해  stack 내부에 있는 값들을 더해준다.