[Baekjoon]백준 10773 제로(실버 4) - Python

문제설명

예제와 설명

문제를 살펴보면 0을 입력받으면 장부에서 제일 최근에 들어온 숫자를 지우고,

그게 아니라면 스택 역할을 수행하는 리스트에 저장하면 된다

처음 n을 입력받을때에는 한 번만 입력받으므로 input()을 사용하고

여러 번 반복하여 입력받을 때는 sys.stdin.readline()을 사용하여 동작 시간을 줄여주도록 코드를 작성하였다

입력받을 때마다 int형으로 변환하여 추후에 합을 구하기 쉽도록 작성하였다

import sys

diary = []
n = int(input())

for _ in range(n):
    money = int(sys.stdin.readline())
    if money == 0:
        diary.pop()
    else:
        diary.append(money)
        
print(sum(diary))

이번 문제는 어렵지 않게 한 번에 정답임을 확인할 수 있다