2022.02.11 [백준] (python 파이썬) 최대 힙

 

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

11279번: 최대 힙

 

---

풀이

파이썬에서는 힙을 만들면 최소 힙이 만들어 진다. 그렇기 때문에 최대 힙을 만들기 위해서는 각 인덱스 값에 -1을 곱해 힙을 만들고 heappop할 때, 다시 -1을 곱해줘 최대 값을 찾는 방식으로 문제를 해결했다.

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