[백준] 10773번 제로(with python)

10773번: 제로

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

www.acmicpc.net

1. 내가 작성한 코드

n = int(input())
arr = []

for i in range(n):
  i = int(input())
  if i != 0:
    arr.append(i)
  elif i == 0:
    arr.pop()

print(sum(arr))

스택에 대해 알고 있다면 손쉽게 풀 수 있는 문제이다.

input된 값이 0이 아니면 append, 0이면 pop을 해주면 된다.

개발자 무무