[백준] 1927번: 최소 힙

Hint

1. 힙 자료구조 이용하기 (힙이란?)
2. 힙에서 heappop()을 하면 가장 작은 값이 반환 됨

 

백준 1927번: 최소 힙

import heapq
import sys

n = int(sys.stdin.readline())

heap = []       # 힙 만들기(리스트랑 만드는 법은 똑같음)
for i in range(n):
    num = int(sys.stdin.readline())
    if num == 0:    # input 숫자가 0 이면 현재 힙에 들어있는 최소값 출력하기
        print(heapq.heappop(heap) if heap != [] else 0)     # 힙이 비어있지 않을 때만 heappop() 하기
    else:
        heapq.heappush(heap, num)   # 힙에 숫자 넣기