카테고리 없음

[백준/파이썬][Gold II] 보석 도둑 - 1202

난쟁이 개발자 2025. 3. 2. 16:11
반응형

[Gold II] 보석 도둑 - 1202

문제 링크

성능 요약

메모리: 172512 KB, 시간: 824 ms

분류

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

제출 일자

2025년 3월 2일 16:09:07

문제 설명

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

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

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

입력

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

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

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

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

출력

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

풀이

import sys
input = sys.stdin.readline

import heapq

def sol_1202() :
    N, K = map(int, input().split())
    jewels = []
    bags = []

    for _ in range(N) :
        m, v = map(int, input().split())
        heapq.heappush(jewels, (m, v))  # 힙에 푸시

    for _ in range(K) :
        bags.append(int(input()))       # 가방 무게 정보
    bags.sort()

    total_val, capable_jewels = 0, []
    for bag in bags :
        while jewels and bag >= jewels[0][0] :  # bag의 무게가 jewels의 무게보다 더 클 경우
            m, v = heapq.heappop(jewels)        # 힙에 팝
            heapq.heappush(capable_jewels, -v)  # 힙에 푸시 (-로 함으로써 큰 값으로 정렬)
        if capable_jewels :
            total_val -= heapq.heappop(capable_jewels)  # total_val 갱신 (다시 - 함으로써 + 로 바꿔서 갱신)
        elif not jewels :
            break

    print(total_val)

sol_1202()

백팩 문제라서 DP로 접근을 했었다. 결국 다른 분들의 풀이를 참고하여 우선순위큐를 활용하여 문제를 해결할 수 있었다. 

반응형