알고리즘 - Python / 백준 - 1927번 : 최소 힙

1927번: 최소 힙 (acmicpc.net)

1927번: 최소 힙

 

---

풀이

• 최소힙을 구현만 하면 되는 문제다.
• 파이썬에서 최소힙은 heapq 모듈을 통해 쉽게 구현할 수 있다.
• 구현만 하고 입력에 따라 push , pop 을 진행하면 된다.

 

import sys
import heapq

heap = []
n = int(sys.stdin.readline())
for _ in range(n):
  m = int(sys.stdin.readline())
  if m == 0:
    if len(heap) == 0:
      print(0)
    else:
      print(heapq.heappop(heap))
  else:
    heapq.heappush(heap, m)

---