최대 힙[백준 11279] - python

 

최대 힙을 이용하여 구현하는 문제이다.

정답코드 먼저 살펴보자

import heapq
import sys

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

for i in range(n) :
  num = int(sys.stdin.readline())
  if num == 0 and len(heap) == 0:
    print(0)
  elif num == 0 :
    print(heapq.heappop(heap)[1])
  else :
    heapq.heappush(heap, (-num, num ))

 

문제에 나와있듯이 힙을 이용해야 겠다는 생각이 들었고

우선순위를 큰 값이 가장 먼저 나올수 있도록 -를 곱해서 거꾸로 넣어주었다.

python에서 heapq를 사용하는 방법은 따로 포스팅을 할 예정이다.

 

 

 

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

11279번: 최대 힙