(Python) - BOJ(1927번) : 최소 힙

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

1927번: 최소 힙

---

• 해설 : 

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

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

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

 

---

• 풀이 :

Python의 heapq 모둘을 사용하여 힙을 사용하면 된다. 

 

 

 

---

• 소스코드 : 

 

import sys
import heapq
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))
            else:
                print(0)
        else:
            heapq.heappush(heap, N)