[BOJ] #1927 최소 힙 (Python)

[BOJ] #1927 minimum heap (Python)

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

1. Python의 내장 모듈인 heapq를 사용하여 문제를 해결할 수 있다. (구현으로도 해결 가능)
2. input이 0이 아닐때: heapq.heappush(리스트, 데이터)로 push
3. input이 0일때: 리스트가 0인지 확인하고 heapq.heappop(리스트) 로 pop후 출력

import heapq
import sys
 
H = []
n = int(sys.stdin.readline())
for i in range(n):
    k = int(sys.stdin.readline())
    if k == 0:
        if len(H) == 0:
            print(0)
            continue
        else:
            print(heapq.heappop(H))
    else:
        heapq.heappush(H, k)

1. line 5, line 7 - 데이터를 input()으로 받는 과정에서 시간 초과 발생
   • python에서 input()보다는 sys 모듈을 활용하여 readline해주는 방법 사용할 것

• heapq 모듈을 사용하여 최소 힙을 구현할 수 있다.
• heapq.heappush(리스트, 데이터)로 push
• heapq.heappop(리스트)로 pop
• 리스트가 비어있을 때 pop을 시도하면 IndexError가 발생하므로 예외처리를 해주어야 한다.
• 시간복잡도: O(NlogN)