[백준]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)

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

출력

#include<bits/stdc++.h>

using namespace std;
typedef long long ll;
typedef pair<ll, ll> pl;
ll  n, k, m, v, sum = 0, idx = 0;

vector<ll>bags;
vector<pl>jews;
priority_queue<ll,vector<ll>>pq;
int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	ll  m, v, c;
	cin >> n >> k;
	for (int i = 1; i <= n; i++) {
		cin >> m >> v;
		jews.push_back({ m,v });
	}
	for (int i = 0; i < k; i++) {
		cin >> c;
		bags.push_back(c);
	}
	sort(jews.begin(), jews.end()); //무게별 소트
	sort(bags.begin(), bags.end());//가방 무게 소트
	
	for (int i = 0; i < bags.size(); i++) { //모든 가방에 대해서..
		for (idx; idx < jews.size(); idx++) { //가방의 무게보다 작은 보석들은 다 우선순위 큐에
			auto [m, v] = jews[idx];
			if (bags[i] >= m) { 
				pq.push(v);
			}
			else break; //만약 가방보다 무거운 보석이면 break.. 더이상 큐에 넣을게 없음
		}
		if (!pq.empty()) { //만약 비어있지 않다면 하나 꺼낼거임
			sum += pq.top(); pq.pop();
		}
	}
	cout << sum;
}

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

 

 

그리디하게 하나씩 빼면서 최적의 요소를 찾아야 한다면 힙을 사용하자..