[백준] 1927번: 최소 힙 - python

https://www.acmicpc.net/problem/1927

1927번: 최소 힙

 

힙 자료구조를 잘 사용할 수 있는지 테스트하는 문제 ^^

 

전에 회의실 문제에서 사용해서 익숙해서 쉽게 풀었다.

import heapq
import sys
input = sys.stdin.readline
n = int(input())

hq = []
for i in range(n):
    m = int(input())
    if m:
        heapq.heappush(hq, m)
    else:
        if hq:
            pop = heapq.heappop(hq)
            print(pop)
        else:
            print(0)

 

완성