[Python] 백준 1927. 최소힙

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

1927번: 최소 힙

풀이 설계:

• 입력받은 값이 0이고
  • 힙이 비어있으면 0을 출력한다.
  • 힙이 비어있지 않으면 값을 꺼낸다.
• 입력받은 값이 0이 아니면 힙에 값을 추가한다.

 

import sys
import heapq

input = sys.stdin.readline
n = int(input())
heap = []

for case in range(n):
    num = int(input())
    if num == 0:
    	#입력받은 값이 0이고
		#힙이 비어있으면 0을 출력한다.
        if len(heap) == 0:
            print(0)
        # 힙이 비어있지 않으면 값을 꺼낸다.
        else:
            print(heapq.heappop(heap))
	# 입력받은 값이 0이 아니면 힙에 값을 추가한다.
    else:
        heapq.heappush(heap, num)