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

https://www.acmicpc.net/problem/10773

10773번: 제로

 

 

import sys
n = int(sys.stdin.readline())
number = [ int(sys.stdin.readline()) for _ in range(n) ]
stack = []
for i in range(len(number)):
    if stack and number[i] == 0: #!
        stack.pop()
    else: #2
        stack.append(number[i])
print(sum(stack)) #3

 

#1 : stack이 비어있지 않고 입력값이 0이라면 stack.pop() 실행

#2 : 아니라면 stack에 값 추가

#3 : 남은 stack의 값 모두 합해서 출력

 

파이썬은 리스트의 합을 구할 때 sum함수를 사용할 수 있어서 편한 것 같다.