[백준 1927 최소 힙] PYTHON

 

문제 이름도 최소 힙이고 쓰라고 써있으니 python 최소 힙이라 검색을 해보자

heapq라는 모듈이 있다. 갖다 쓰자

아래처럼 간단하게 짤 수 있다.

 

혹시 모르시는 분들을 위해 heapq에 대해 설명을 하면

heapq.heappop(Q)는 최소 값을 pop하고 반환한다.

heapq.heappop(Q,i)는 i를 heapq에 넣는 것이다.

참고로 최소 힙의 정의는 부모 노드의 키값이 자식 노드의 키값보다 항상 작은 힙이다.

 

+ sys.stdin.readline()은 input 값이 많을 때 유용하다 (시간적으로)

 

 

import heapq
import sys

a=int(input())
heap=[]
for _ in range(a):
    i = int(sys.stdin.readline())
    if(i==0):
        try:
            print(heapq.heappop(heap))
        except:
            print(0)
    else:
        heapq.heappush(heap,i)