[백준/파이썬] 14235번

📍백준 14235번 문제

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

코드

import heapq
import sys
input = sys.stdin.readline

n = int(input())
que = []
for _ in range(n):
    temp = list(map(int, input().split()))
    if len(temp) == 1 and temp[0] == 0:
        if que:
            print(heapq.heappop(que)[1])
        else:
            print(-1)
    for i in temp[1:]:
        heapq.heappush(que, (-i, i))

풀이

최대힙을 이용해서 문제를 풀면 된다.

• 만약 입력받은 값이 0이라면, 아이에게 선물을 주어야 하므로 heappop을 통해 선물을 출력하면 된다.
• 0이 아닌 선물 리스트를 입력받았다면, 선물의 개수인 첫번째 값을 제외한 값들을 que에 저장하면 된다.