[백준] 11279번 최대 힙 - 파이썬

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

1463번: 1로 만들기

풀이

저번에 풀었던 문제와 다른 최대 힙입니다. heapq는 최소 힙이기에 부호를 역전시켜서 힙에 넣어준뒤에 꺼내 줄때는 다시 부호를 재역전시켜줍시다.

# 11279번 최대 힙
import sys, heapq

heap=[]
n = int(input())
for _ in range(n):
    k = int(sys.stdin.readline())
    if k == 0:
        if heap:
            print((-1)*heapq.heappop(heap))
        else:
            print(0)
    else:
        heapq.heappush(heap, -1*k)

출력결과