Koala - 6기/기초 알고리즘 스터디

[백준/Python] 10773 제로

항붕쿤 2022. 3. 20. 16:14

​​문제



문제풀이

 


문제의 경우 0이 나오면 앞에 나온 숫자를 지우면 되기 때문에 파이썬에서는 이를 리스트를 이용하여 쉽게 구현해 낼 수 있다. 먼저 변수 a에 숫자를 입력받고 입력받은 숫자만큼 반복문을 돌려 b를 리스트에 삽입한다. 그 이후 b에 0이 나올 때마다 리스트에 가장 나중에 들어온 즉, 맨 뒤의 리스트 요소를 삭제시키고 마지막에 리스트 요소의 총 합을 출력한다. 

 

import sys


a = int(sys.stdin.readline())
li = []

for i in range(a):
    b = int(sys.stdin.readline())
    if b == 0:
        del li[-1]
    else:
        li.append(b)
print(sum(li))