백준 10773 (제로) [Python/파이썬] [Implement/구현]

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

10773번: 제로

 

---

나:

import sys

input = sys.stdin.readline

k = int(input())
money = []
for i in range(k):
    a = int(input())
    if a== 0:
        money.pop(-1)
    else:
        money.append(a)

print(sum(money))

완성된 코드는 위와 같다.

 

(1)
import sys

input = sys.stdin.readline

k = int(input())
money = []

첫째 줄에 정수 k를 입력받고 다음중에는 입력값들이 들어갈 리스트 money 를 선언한다.

이때 sys 모듈을 이용해 input() 함수를 이용한다. 왜냐하면 그냥 input()만 사용시 채점할 때 '시간초과'가 떠서 sys를 사용하면 대부분의 시간초과 문제를 해결할 수 있다.

 

(2)
for i in range(k):
    a = int(input())
    if a== 0:
        money.pop(-1)
    else:
        money.append(a)

print(sum(money))

입력받은 k 만큼 반복문 for를 사용해 0 이면 리스트에 차례대로 넣은 수들 중에 젤 마지막 요소를 pop()을 이용해 빼고 그 외의 경우는 리스트에 순서대로 입력값을 추가시켰다. 

최종적으로 리스트 요소들의 합을 sum()을 이용해 출력시킨다.