문제
문제풀이
문제의 경우 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))
'Koala - 6기 > 기초 알고리즘 스터디' 카테고리의 다른 글
[백준/python] 2675번: 문자열 반복 (0) | 2022.03.20 |
---|---|
[백준/Python] 14910번: 오르막 (0) | 2022.03.20 |
[백준/파이썬]10808번 알파벳 갯수 (0) | 2022.03.20 |
[백준/c++] 17094번: Serious Problem (0) | 2022.03.20 |
[백준/python] 5363번: 요다 (0) | 2022.03.20 |