[백준/파이썬] 11279번 최대 힙

import heapq
from sys import stdin
n = int(stdin.readline())
heap = []
for i in range(n):
    num = int(stdin.readline())
    if num == 0:
        if len(heap) == 0:
            print(0)
        else:
            print(heapq.heappop(heap) * -1)
    else:
        heapq.heappush(heap, -num)

처음에 리스트를 통해서 해결을 했으나 시간 초과가 생겨서 찾아보니 힙을 이용하는 것이어서 아래 블로그를 참고하여 만들었습니다.

 

참고: https://redcrow.tistory.com/487