[BOJ] 1202. 보석 도둑 ✔

1202. 보석 도둑

1. 문제

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

상덕이가 털 보석점에는 보석이 총 N개 있다. 각 보석은 무게 $M_i$와 가격 $V_i$를 가지고 있다. 상덕이는 가방을 K개 가지고 있고, 각 가방에 담을 수 있는 최대 무게는 $C_i$이다. 가방에는 최대 한 개의 보석만 넣을 수 있다.

상덕이가 훔칠 수 있는 보석의 최대 가격을 구하는 프로그램을 작성하시오.

2. 입력

첫째 줄에 N과 K가 주어진다. (1 ≤ N, K ≤ 300,000)

다음 N개 줄에는 각 보석의 정보 $M_i$와 $V_i$가 주어진다. (0 ≤ $M_i$, $V_i$ ≤ 1,000,000)

다음 K개 줄에는 가방에 담을 수 있는 최대 무게 $C_i$가 주어진다. (1 ≤ $C_i$ ≤ 100,000,000)

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

3. 출력

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

4. 풀이

C++

#include <iostream>
#include <queue>
#include <vector>
#include <utility>
#include <algorithm>
#define ioboost ios::sync_with_stdio(false); cout.tie(NULL); cin.tie(NULL);
using namespace std;
typedef pair<int, int> P;

int N, K;
vector<P> jewel;
vector<int> beg;

bool cmp(const P &a, const P &b)
{
    return a.first < b.first;
}

void Input()
{
    cin >> N >> K;
    
    int M, V, C;
    // M: 보석의 무게, V: 보석의 가치, C: 가방의 한도
    for (int i = 0; i < N; ++i)
    {
        cin >> M >> V;
        jewel.push_back(make_pair(M, V));
    }    
    
    for (int i = 0; i < K; ++i)
    {
        cin >> C;
        beg.push_back(C);
    }   
}

void Solve()
{
    sort(jewel.begin(), jewel.end(), cmp); // 보석 무게를 기준으로 오름차순 정렬
    sort(beg.begin(), beg.end()); // 가방에 담을 수 있는 무게를 기준으로 오름차순 정렬
    
    priority_queue<int> pq;
    
    long long ans = 0;
    int idx = 0;
    
    // 한도가 낮은 가방부터 탐색
    for (int i = 0; i < K; ++i)
    {
    	// 해당 가방에 넣을 수 있는 모든 보석의 가치를 우선순위 큐에 저장
        while (beg[i] >= jewel[idx].first && idx < N)
        {
            pq.push(jewel[idx].second);
            ++idx;
        }
        
        // 그 중 가장 가치가 높은 보석을 해당 가방에 할당
        if (!pq.empty())
        {
            ans += pq.top();
            pq.pop();
        }
    }
    
    cout << ans << '\n';
}


int main()
{
    ioboost;
    Input();
    Solve();
}