본문 바로가기

코딩테스트/BOJ

[백준/파이썬] 10773번: 제로

문제

www.acmicpc.net/problem/10773

 

10773번: 제로

첫 번째 줄에 정수 K가 주어진다. (1 ≤ K ≤ 100,000) 이후 K개의 줄에 정수가 1개씩 주어진다. 정수는 0에서 1,000,000 사이의 값을 가지며, 정수가 "0" 일 경우에는 가장 최근에 쓴 수를 지우고, 아닐 경

www.acmicpc.net

풀이

 - 구현

 

 - 리스트의 append()와 pop()이용

  • [1] append
  • [1,3] append
  • [1,3,5] append
  • [1,3,5,4] append
  • [1,3,5] (0을 불렀기 때문에 최근의 수를 지운다) pop
  • [1,3] (0을 불렀기 때문에 그 다음 최근의 수를 지운다) pop
  • [1,3,7] append
  • [1,3] (0을 불렀기 때문에 최근의 수를 지운다) pop
  • [1] (0을 불렀기 때문에 그 다음 최근의 수를 지운다) pop
  • [1,6] append

 

 

코드

k = int(input())
arr = [0]
for i in range(k):
    n = int(input())
    if n == 0:
        arr.pop()
    else:
        arr.append(n)

print(sum(arr))