[우선순위 큐/Python] 11279번: 최대 힙 - 효과는 굉장했다!

11279번: 최대 힙

 

알고리즘 분류

• 자료 구조
• 우선순위 큐

 

 

SOLUTION

import sys
import heapq

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

for _ in range(N):
    x = int(sys.stdin.readline())
    
    if x == 0 :
        if heap: # heap이 비어있지 않다면
            print(heapq.heappop(heap)[1]) # heap[1]이 -를 붙히지 않은 원래 값이므로

        else:
            print(0)

    else:
        # heappush는 가장작은 값이 0번째 index로 들어가기때문에 -를 붙혀서 가장 큰 값을 맨 앞에 넣도록 만든다
        heapq.heappush(heap, [-x, x])