[백준 1927번] 최소 힙

import heapq
import sys

heap = []

N = int(input())
for _ in range(N):
    X = int(sys.stdin.readline())
    if X == 0:
        if len(heap) == 0:
            print("0")
        else:
            print(heapq.heappop(heap))
    else:
        heapq.heappush(heap, X)

힙을 이용해서 풀었습니다.

처음에 X의 입력을 input()으로 구현했는데 시간초과가 나서

stdin을 이용해 입력 받았더니 통과하였습니다.