[백준] 1927. 최소힙 / 11279. 최대힙 / python 파이썬

🚩 heap, 자료구조, 우선순위 큐

 

thinking

input이 10만줄 이상으므로 sys.stdin.readline() 을 사용했고,

파이썬의 heapq 모듈을 사용하여 해결했다.

 

코드

1927. 최소힙

import sys
import heapq

N = int(sys.stdin.readline())
heap = []
for _ in range(N):
    x = int(sys.stdin.readline())
    if x == 0:
        if heap:
            print(heapq.heappop(heap))
        else:
            print('0')
    else:
        heapq.heappush(heap, x)

 

11279. 최대힙

heapq 모듈은 기본적으로 최소힙만을 지원하기 때문에 최대힙을 구하려면 튜플이나 리스트형식으로 넣어 0번째 인덱스의 값을 기준으로 최소힙을 구성해야한다.

import sys
import heapq

N = int(sys.stdin.readline())
heap = []
for _ in range(N):
    x = int(sys.stdin.readline())
    if x == 0:
        if heap:
            print(heapq.heappop(heap)[1])
        else:
            print('0')
    else:
        heapq.heappush(heap, [-x,x])