Algorithm Problems

[백준] [Python] 10773번 제로 - 스택

WakaraNai 2021. 4. 24. 22:50
728x90
반응형

www.acmicpc.net/problem/10773

 

10773번: 제로

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

www.acmicpc.net

 

Python

import sys

stack = []
for _ in range(int(sys.stdin.readline())):
    x = int(sys.stdin.readline())
    if x == 0: del stack[-1]
    else: stack.append(x)
    
print(sum(stack))

 

 

728x90
반응형