[Python] 백준 14235번 - 크리스마스 선물

📖 문제

14235번: 크리스마스 선물

 

🔎 풀이

우선순위 큐를 활용하는 문제입니다. 특정한 조건에 따라 우선순위가 가장 높은 요소를 추출합니다.

파이썬의 heapq 모듈은 최소 힙만을 지원합니다. 하지만 음수 형태로 저장한다면, 최대 힙처럼 활용이 가능합니다.

 

💻 코드

import sys
import heapq
input = sys.stdin.readline

# 선물을 담을 배열
giftbox = []
for _ in range(int(input())):
    a = list(map(int, input().split()))
    # 입력값이 0인 경우 선물 주기
    if a[0] == 0:
        print(-heapq.heappop(giftbox) if len(giftbox) else -1)
    else:
        # 선물 박스에 선물 담기
        # 파이썬은 min heap만 지원하기 때문에 -를 붙혀 max heap처럼 사용
        for i in range(a[0]):
            heapq.heappush(giftbox, -a[i+1])