[C++]/백준 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

3 2
1 65
5 23
2 99
10
2

• 예제 출력

10

164

• 접근방식

우선, M과 V를 저장하는 벡터를 준비하고 가방의 최대 갯수를 저장하는 벡터를 준비한다.
두개의 벡터를 오름차순으로 정렬하고 가방에 들어갈 수 있는 보석을 찾고 PQ(우선순위 큐)에 넣어준다.
우선순위 큐는 내림차순으로 설정해주면 젤 위에 있는 값이 가장 가격이 높은 보석이기때문에 top만 결과에 넣어주고 pop하면 된다.
여기서 범위가 Ci의 경우 100,000,000까지 이기 때문에 long long 자료형을 사용해야한다.

• 코드

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

using namespace std;

long long n, m, v, k, c, result;
vector<pair<int, int>> v1;
vector<int> v2;

int main()
{
    cin >> n >> k;

    for (int i = 0; i < n; i++)
    {
        cin >> m >> v;
        v1.push_back({ m,v });
    }

    for (int i = 0; i < k; i++)
    {
        cin >> c;
        v2.push_back(c);
    }

    sort(v1.begin(), v1.end());
    sort(v2.begin(), v2.end());
    priority_queue<int> pq;
    int j = 0;
    for (int i = 0; i < k; i++) 
    {
        while (j < n && v1[j].first <= v2[i])
        {
            pq.push(v1[j++].second);
        }
        if (pq.size())
        {
            result += pq.top();
            pq.pop();
        }
    }
    cout << result;
    return 0;
}

• 문제링크 : https://www.acmicpc.net/problem/1202

1202번: 보석 도둑