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))
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))