(Python) - BOJ(11279번) : 최대 힙

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

11279번: 최대 힙

---

• 해설 : 

널리 잘 알려진 자료구조 중 최대 힙이 있다. 최대 힙을 이용하여 다음과 같은 연산을 지원하는 프로그램을 작성하시오.

 

1. 배열에 자연수 x를 넣는다.
2. 배열에서 가장 큰 값을 출력하고, 그 값을 배열에서 제거한다.

 

프로그램은 처음에 비어있는 배열에서 시작하게 된다.

 

 

 

---

• 풀이 :

heapq 모듈을 사용하여 해결할 수 있다. 그러나 heapq의 기본형이 minheap이므로 인자에 역수를 취해주어 maxheap으로 사용해야 한다.

 

 

 

---

• 소스코드 : 

 

import heapq
import sys
input = sys.stdin.readline
 
if __name__ =="__main__":
    heap = []
    for _ in range(int(input())):
        N = int(input())
        if N == 0:
            if heap:
                print(heapq.heappop(heap)[1])
            else:
                print(0)
        else:
            heapq.heappush(heap, (-N,N))