[파이썬, Python] 백준 1927: 최소 힙

문제

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

1927번: 최소 힙

코드

import sys
import heapq
input = sys.stdin.readline

n = int(input())
heap = []

for i in range(n):
    a = int(input())
    
    if a != 0:
        heapq.heappush(heap, a)
    else:
        if not heap:
            print("0")
        else:
            print(heapq.heappop(heap))

설명

처음에 모듈 없이 단순 구현만으로 풀었을 때, 시간초과가 나서 검색을 통해 풀이 방법을 알았다.

heapq라는 모듈이 필요하다. 이 모듈을 통해 push를 할 때, 정렬이 된 상태로 push를 할 수 있다.

따라서 heap이라는 리스트를 만들어주고, 모듈을 통해 수를 push, pop 해주면서 조건에 맞게 출력하면 되는 문제였다.