[백준 1927, Silver II] 최소 힙

📒문제

📢풀이

import sys
import heapq

q = []
N = int(input())

for _ in range(N):
    num = int(sys.stdin.readline())

    if num == 0:
        if q:
            print(heapq.heappop(q))
        else:
            print("0")
    else:
        heapq.heappush(q, num)

처음에 heapq 모듈을 썼는데 시간초과가 떠서 의아했었던 문제다. 힙을 실제로 구현해야 되나 생각까지 했었는데, 그럴 필요까지는 없고 입력을 받을 때 sys.stdin.readline()을 사용했어야지만 통과할 수 있는 문제였다. input()으로는 통과가 불가했다.

앞으로 코딩테스트 문제를 풀때 input대신 sys.stdin.readline()을 사용하는 습관을 들여두자!