백준 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 1 5 10 100 100 11 | 10 |
| 3 2 1 65 5 23 2 99 10 2 | 164 |
접근 방식
: 가방 리스트에 대해 보석 리스트를 돌면서 해당 가방에 넣을 수 있는 보석을 찾아 가격을 더하고 pop 반복. 가방 오름차순 정렬, 보석 가격 기준 내림차순 정렬
---> 틀렸습니다.
알게된 점
- 용량이 작은 가방부터 보석을 담는다. --> 가방 무게 리스트 오름차순 정렬. 보석 리스트 오름차순 정렬(무게 기준)
- 각 용량에 대하여 담을 수 있는 모든 보석을 바구니에 넣는다.
- 바구니에서 가장 가격이 큰 것을 바구니에서 빼서 가방에 넣는다. ---> 우선순위 큐(힙) 사용. 최대힙
코드
import sys
import heapq
n, k = map(int, sys.stdin.readline().split())
jws = []
for _ in range(n):
m, v = map(int, sys.stdin.readline().split())
jws.append((m, v))
wei = []
for _ in range(k):
wei.append(int(sys.stdin.readline()))
wei.sort()
jws.sort()
q = []
ans = 0
for i in range(len(wei)):
j = 0
while jws and wei[i]>=jws[0][0]:
heapq.heappush(q, -heapq.heappop(jws)[1])
if q:
ans -= heapq.heappop(q)
elif not jws:
break
print(ans)