[백준 11279번] 최대 힙 - python

문제 링크

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

11279번: 최대 힙

풀이

- 파이썬은 최대힙을 지원하지 않기 때문에 최소 힙에 삽입할 때 마이너스를 붙여서 삽입하여 구현한다. 

import heapq
import sys
input = sys.stdin.readline 

n = int(input())

h = []
for _ in range(n):

  x = int(input())

  if x == 0:
    if h:
      # 절댓값 출력 
      print(abs(heapq.heappop(h)))
    else:
      print(0)
  else:   
    # 마이너스 부호 붙여서 삽입 
    heapq.heappush(h,-x)