[백준] 10773 제로 python 알고리즘 문제

문제 10773. 제로

1. 나의 코드와 발상 과정

import sys

k = int(sys.stdin.readline())
money_list = []

for i in range(k):
    a = int(sys.stdin.readline())
    if a == 0:
        money_list.pop()
    else:
        money_list.append(a)

print(sum(money_list))

문제도 어렵지 않았고, 코드 구성도 간단했다. 

 

money_list 속에 a를 받아오는데, 만약 0이라면 현재 입력값을 넣지 않고, 그 전 반복에서 넣었던 숫자를 뺀다.

a가 0이 아닐 경우에만 계속 list에 집어 넣고, money_list의 전체 입력값을 더해 출력한다.

 

 

 

 

 

문제 출처:

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