[Python] BOJ 10733: 제로

Binsu·2021년 8월 14일
0

Algorithms

목록 보기
6/22

문제

풀이 방법

K = int(input())
stack = []

for _ in range(K):
    num = int(input())
    if num == 0:
        stack.pop()
    else:
        stack.append(num)
print(sum(stack))

생각보다 간단한 문제였다. num으로 들어온 정수가 0일 때, pop() 함수만 활용하면 쉽게 스택(FILO 구조)에서 Last-In 원소가 pop 된다.
CPython은 시간 초과로 실패하여 PyPy로 제출하였다. input() 대신 sys.stdin.readline()을 사용하면 통과하지 않을까 싶다.

0개의 댓글