[python] 백준 - 14235. 크리스마스 선물

🤔문제 해결

• S2 | 우선순위 큐

우선순위 큐 문제, 파이썬의 힙큐 모듈을 사용하자.

힙큐 모듈은 최소값을 반환하는 최소힙을 기본으로 한다.

따라서 음수로 값을 넣고 값을 뽑아서 다시 음수를 곱해주면 최대값이 된다.

 

1. 인풋값이 0일 때
   1. 가지고 있는 선물이 있으면 선물을 나눠준다.
   2. 없으면 꽝
2. 인풋값이 0이 아닐 때
   1. 선물을 충전한다.

 

 

 

💻소스 코드

import heapq

if __name__ == '__main__':
    N = int(input())
    pq = []  # 선물을 담을 리스트 ( 우선순위 큐 )
    for _ in range(N):
        input_value = input()
        if input_value == '0':  # 인풋값이 0일 때
            if pq:  # 선물이 있으면 선물
                print(-heapq.heappop(pq))
            else:  # 선물이 없으면 -1
                print(-1)

        else:
            # 거점에서 선물 충전
            present_list = list(map(int, input_value.split()))
            for i in range(1, present_list[0] + 1):
                heapq.heappush(pq, -present_list[i])
 

 

📕문제 확인

출처: BACKJOON ONLINE JUDGE

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

14235번: 크리스마스 선물