백준 1202번 보석도둑

 

 

문제

문제링크

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

상덕이가 털 보석점에는 보석이 총 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)

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

출력

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

2. 코드

import sys
from heapq import heappush, heappop

input = sys.stdin.readline

n, k = map(int, input().split())

b = [list(map(int, input().split())) for _ in range(n)]
bag = [int(input()) for _ in range(k)]

# 정렬를 통해서 가방 무게 에 맞게 가격을 최대화 시키기 위함
b.sort()
bag.sort()

ans = []
result = 0
# 가방 무게
for w in bag:
    # 만약 보석들이 존재 하면서 가방의 무게가 보석의 무게보다 더 크면
    while b and w >= b[0][0]:
        # 최대힙으로 보석의 가치를 넣어줌
        heappush(ans, -b[0][1])
        # 가방에 넣어준 보석을 뺌
        heappop(b)
    # while에 있는 조건문 true 였다면 진행
    if ans:
        result += heappop(ans)
    # 보석들이 남아 있으면 게속 진행
    elif b:
        continue
print(-result)

3. 회고

Q. 문제를 보고 든 생각

• 입력 란에 N, K 범위를 보고 이중 포문이 아닌 다른 방식으로 풀어야 겠다는 생각이 들었다
• 보석의 최대가격을 원하기에 최대 힙을 이용해 당장의 욕심을 해결해나가는 그리디를 풀어 나가야겠다는 생각을 했다

4. 고쳐야 할점

최대 힙, 최소 힙을 짜는 연습을 해야겠다