본문 바로가기

Programming [Python]/백준 알고리즘 솔루션

#106 백준 파이썬 [1927] 최소 힙

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

 

#Solution

파이썬 heapq 모듈을 사용하면 쉽게 구할 수 있다. 힙 정렬에 대한 이해도가 된 상태에서 사용하는 걸 권장! 시간복잡도는 O(log2n)

import sys
import heapq

numbers = int(input())
heap = []

#Max Heap
for _ in range(numbers):
    num = int(sys.stdin.readline())
    if num != 0:
        heapq.heappush(heap, num)
    else:
        try:
            print(heapq.heappop(heap))
        except:
            print(0)