[baekjoon] 백준 1927번 : 최소 힙 (by python 파이썬) with heapq

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

1927번: 최소 힙

 

정답 code

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

 

이번 문제를 풀기위해 힙 모듈을 공부해야 했다.

 

참고한 사이트는 아래와 같다.

https://www.daleseo.com/python-heapq/

[파이썬] heapq 모듈 사용법

 

힙은 아래와 같이 리스트에 값을 넣고

heapq.heappush(리스트, 값 전달 인자)

아래와 같이 출력하면 가장 작은수를 삭제하고 리턴한다.

print(heapq.heappop(리스트))