알고리즘 - Python / 백준 - 11279번 : 최대 힙

11279번: 최대 힙 (acmicpc.net)

11279번: 최대 힙

 

---

풀이

• 최대힙을 구현만 하면 되는 문제이다.
• 최대힙은 최소힙 구현에서 입력되는 수들에 -1 만 곱해주는 방식으로 간단하게 구현할 수 있다.
• 전 문제에서 만든 최소힙을 활용해 입력되는 수들에 push , pop 단계에 -1 만 곱해주도록 하면 된다.

 

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((-1)*heapq.heappop(heap))
  else:
    heapq.heappush(heap, (-1)*m)

---