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

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

14235번: 크리스마스 선물

 

문제

 

풀이

우선순위 큐를 활용해서 가치가 가장 높은 선물부터 아이들에게 나눠줄 수 있게 코드를 짰습니다.

 

 

코드

# 14235번 크리스마스 선물
from heapq import heappush, heappop
n = int(input())
values = []

for _ in range(n):
    a = list(map(int, input().split()))
    if a[0] == 0:
        # 아이를 만났고 선물이 있는 경우
        if values:
            print(heappop(values)[1])
        # 아이를 만났지만, 선물이 없는 경우
        else:
            print(-1)
    else:
        for i in range(1, a[0]+1):
            heappush(values, [-a[i], a[i]])

 

 

출력결과