[백준] 1927. 최소 힙

원숭2·2022년 1월 26일
0

백준

목록 보기
14/54

문제

풀이

  1. heapq 모듈을 사용함.
  2. heapq 모듈은 list를 최소 힙 처럼 다룰 수 있도록 도와줌.
  3. 0이 입력될 때, list가 비어 있으면 0을 출력하고, 아니면 heap에서 원소를 삭제 후 출력함.
  4. 그 외의 수가 입력될 시 heap에 추가해줌.

코드

import heapq
import sys

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

solution()

0개의 댓글