[코테 - py] 최대 힙

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

 

오늘은 어제에 이어서 최대 힙에 관한 문제였다.

이것도 heapq를 사용하고, 힙에 push 할때와 pop할 때 (-)를 붙여주면 최소힙을 최대힙처럼 사용할 수 있다.

 

import sys
import heapq

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

 

앞으로도 다양한 문제를 풀어봐야지!ㅎㅎ