[백준 - python] 10773번: 제로

Interrobang 2023. 9. 26. 20:11

문제 링크

10773번: 제로

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

www.acmicpc.net

문제 풀이

import sys
input = sys.stdin.readline

stack = []

k = int(input())
for _ in range(k):
    num = int(input())
    match num == 0:
        case True:
            stack.pop()
        case False:
            stack.append(num)

print(sum(stack))

*key point: match case문으로 입력값이 0 인지 아닌지만 구분하여 처리해주고 최종적으로 수들을 합해주면 된다.