[백준] 11279번: 최대 힙

Hint

1. 힙 자료구조 이용하기 (힙이란?)
2. 힙에서 heappop()을 하면 가장 작은 값이 반환 됨. 그렇다면 가장 큰 값을 반환하려면..?
3. 힙에 넣은 숫자들을 마이너스 해서 넣는다면 heappop() 했을 때 어떤 숫자가 먼저 뽑힐까?

 

 

백준 11279번: 최대 힙

import heapq
import sys

n = int(sys.stdin.readline())

heap = []       # 힙 만들기(리스트랑 만드는 법은 똑같음)
for i in range(n):
    num = int(sys.stdin.readline())
    if num == 0:    # input 숫자가 0 이면 현재 힙에 들어있는 최대대값 출력하기
        print((heapq.heappop(heap))*(-1) if heap != [] else 0)  # 힙이 비어있지 않을 때만 heappop() 하기
    else:
        heapq.heappush(heap, -num)      # 힙에 숫자 넣을 땐 마이너스해서 넣기기