PS/큐(queue)

[백준 파이썬(python) 1927번 문제 최소힙 ]

이거시원조랑께 2023. 8. 14. 08:46
반응형

문제

설명

우선순위 큐 (heapq)를 사용하면 되는 문제, 파이썬에서는 min-heapq가 기본값이기 때문에 그냥 heapq를 쓰면 최소 힙이 된다.

코드

import sys
import heapq

input = sys.stdin.readline

N = int(input())
li = []
for _ in range(N):
    num = int(input())
    if num == 0:
        if li:
            print(heapq.heappop(li))
        else:
            print(0)
    else:
        heapq.heappush(li, num)
반응형