[Python] 백준 11279번 - 최대 힙

📖 문제

11279번: 최대 힙

 

🔎 풀이

우선순위 큐를 활용하는 간단한 문제입니다. 파이썬 모듈 heapq를 이용하여 풀이합니다. 파이썬은 최소 힙만을 지원하기 때문에, 최대 힙 문제를 풀기 위해서는 작은 아이디어 하나가 필요합니다. 힙에 저장할 때, 입력값의 부호를 바꾸어 넣어주는 방식으로 최대 힙 문제를 해결할 수 있습니다.

 

💻 코드

import sys
import heapq
input = sys.stdin.readline

N = int(input())
hq = []
for _ in range(N):
    num = int(input())
    if num == 0:
    	# 비어있는 경우 0 출력
        print(-(heapq.heappop(hq)) if hq else 0)
    else:
        heapq.heappush(hq, -num)