[Python] 백준 #10773- 제로

10773번: 제로

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

www.acmicpc.net

풀이

n = int(input())
a = []
for _ in range(n):
    x = int(input())
    a.pop() if x==0 else a.append(x)
print(sum(a))

스택을 사용하여 풀었다.

x가 0이면 pop을 하고 x가 0이아니면 리스트에 추가한다.

합은 sum을 사용하였다.