[알고리즘] 백준 1927 최소힙 / python

 

import heapq
import sys

n = int(input())
h = []

for _ in range(n):
    num = int(sys.stdin.readline())
    if num == 0:
        if not h:
            print(0)
        else:
            print(heapq.heappop(h))
    else:
        heapq.heappush(h, num)

 

heapq라이브러리를 이용해 힙을 구현
import heapq
heapq.heappush(heap, item)
heapq.heappop(heap)

시간초과 당해서 sys.stdin.readline()으로 바꿔씀!