[백준] Python - 11279 최대 힙

문제 링크

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

11279번: 최대 힙

 

문제 풀이

파이쎤 heapq 모듈만 알고 있으면 쉽게 풀 수 있는 문제였던 것 같다. 주의할 점은 heapq 모듈이 기본적으로 최소힙으로 구현되었기 때문에 최대힙으로 바꿔주기 위해 (-n, n) 처럼 튜플로 만들어서 추가해주는 것이다. 혹은 - 를 곱하여 음수로 저장하는 것도 방법이다!

1. Heap에 (-n, n) 튜플 push()
2. Heap에서 두번째 원소를 pop()

 

전체 코드

import sys
import heapq as hq

input = sys.stdin.readline

num = int(input())
max_heap = []

for i in range(num) :
    n = int(input())
    if n == 0 :
        if len(max_heap) <= 0 :
            print(0)
        else : 
            print(hq.heappop(max_heap)[1])
    else :
        hq.heappush(max_heap, (-n, n))