백준 1202번 : 보석 도둑 (c++)

https://www.acmicpc.net/problem/1202

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)

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

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

예제 입력 1 
3 2
1 65
5 23
2 99
10
2
예제 출력 1 
164

#include <iostream>
#include <queue>
#include <vector>
#include <algorithm>

using namespace std;


int N, K;
vector <pair<int, int>> jew;
vector <int> bag;
priority_queue<int> pq;

int main() {
	ios_base::sync_with_stdio(0);
	cin.tie(0); cout.tie(0);
	freopen("input.txt", "r", stdin);
	
	cin >> N>> K;
	int a, b;
	for (int i = 0; i < N; i++)
	{
		cin >> a >> b;
		jew.push_back({ a,b });
	}
	sort(jew.begin(), jew.end());
	for (int i = 0; i < K; i++)
	{
		cin >> a;
		bag.push_back(a);
	}
	sort(bag.begin(), bag.end());
	
	long long result = 0;
	int j = 0;
	for (int i = 0; i < K; i++)
	{
		while (j < N && jew[j].first <= bag[i]) {
			pq.push(jew[j++].second);
		}
		if (!pq.empty()) {
			result += pq.top();
			pq.pop();
		}
	}
	cout << result;

	return 0;
}