[python/파이썬] 백준 10773 제로

[문제 출처]

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

10773번: 제로

 

 

[소스 코드]

입력이 들어왔을 때 0이 아닌 숫자라면 리스트에 저장하고, 0이라면 가장 최근에 저장된 -1번째 숫자를 제거한다.

그 후에 리스트의 합을 구하면 되는 간단한 문제이다.

#10773
import sys

input = sys.stdin.readline

arr = []
n = int(input())
for i in range(n):
  a = int(input())
  if a == 0:
    del arr[-1]
  else:
    arr.append(a)

print(sum(arr))