728x90
▶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)
모든 숫자는 양의 정수이다.
▶출력
첫째 줄에 상덕이가 훔칠 수 있는 보석 가격의 합의 최댓값을 출력한다.
▶풀이
import heapq
n, k = map(int, input().split())
dia = []
for _ in range(n):
w, v = map(int, input().split())
heapq.heappush(dia, [w, v])
bag = []
for _ in range(k):
c = int(input())
heapq.heappush(bag, c)
total_value = 0
c_dia = []
for i in range(k):
c = heapq.heappop(bag)
while dia and c >= dia[0][0]:
[w, v] = heapq.heappop(dia)
heapq.heappush(c_dia, -v)
if c_dia:
total_value -= heapq.heappop(c_dia)
elif not dia:
break
print(total_value)
728x90
'BOJ Code > Gold' 카테고리의 다른 글
[백준/BOJ] gold4 - 7662번 이중 우선순위 큐 (Python) (0) | 2022.06.03 |
---|---|
[백준/BOJ] gold1 - 11041번 이항 계수 3 (Python) (0) | 2022.04.17 |
[백준/BOJ] gold1 - 1194번 달이 차오른다, 가자. (Python) (2) | 2022.04.03 |
[백준/BOJ] gold4 - 4485번 녹색 옷 입은 애가 젤다지? (Python) (1) | 2022.04.03 |
[백준/BOJ] gold4 - 1600번 말이 되고픈 원숭이 (Python) (0) | 2022.04.02 |