1202. 보석 도둑

업데이트 시간 : 2023-07-14 03:22:19 +0000

[Gold II] 보석 도둑 - 1202

문제 링크

성능 요약

메모리: 200260 KB, 시간: 2200 ms

분류

자료 구조, 그리디 알고리즘, 우선순위 큐, 정렬

문제 설명

세계적인 도둑 상덕이는 보석점을 털기로 결심했다.

상덕이가 털 보석점에는 보석이 총 N개 있다. 각 보석은 무게 M_i와 가격 V_i를 가지고 있다. 상덕이는 가방을 K개 가지고 있고, 각 가방에 담을 수 있는 최대 무게는 C_i이다. 가방에는 최대 한 개의 보석만 넣을 수 있다.

상덕이가 훔칠 수 있는 보석의 최대 가격을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N과 K가 주어진다. (1 ≤ N, K ≤ 300,000)

다음 N개 줄에는 각 보석의 정보 M_i와 V_i가 주어진다. (0 ≤ M_i, V_i ≤ 1,000,000)

다음 K개 줄에는 가방에 담을 수 있는 최대 무게 C_i가 주어진다. (1 ≤ C_i ≤ 100,000,000)

모든 숫자는 양의 정수이다.

출력

첫째 줄에 상덕이가 훔칠 수 있는 보석 가격의 합의 최댓값을 출력한다.

💡 Solutions

📄 보석 도둑.py

import sys
import heapq
# https://www.acmicpc.net/problem/1202
N, K = map(int, sys.stdin.readline().split())

jams = []
for _ in range(N):
    M, V = map(int, sys.stdin.readline().split())
    heapq.heappush(jams, [M, V]) # 무게별로 push

bags = list(int(sys.stdin.readline()) for _ in range(K))
bags.sort()

# print(N,K)
# print(*jams, sep='\n')
# print(bags)

bag_cargo = []
ans = 0
for idx, bag in enumerate(bags):
    while jams and (bag >= jams[0][0]):
        jam = heapq.heappop(jams)
        heapq.heappush(bag_cargo, -jam[1]) # 가중치별로 push
        
    if bag_cargo:
        ans -= heapq.heappop(bag_cargo)
    elif not jams:
        break


print(ans)