백준 1202 보석도둑 풀이(c++)

문제 링크

성능 요약

메모리: 15328 KB, 시간: 156 ms

분류

자료 구조, 그리디 알고리즘, 우선순위 큐, 정렬

제출 일자

2025년 3월 5일 16:38:35

문제 설명

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

상덕이가 털 보석점에는 보석이 총 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 <iostream>
#include <vector>
#include <algorithm>
#include <queue>

using namespace std;

int N, K;
long long iResult = 0;

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL); cout.tie(NULL);
    cin >> N >> K;

    vector<pair<long long, long long>> vJewel(N);
    for (int i = 0; i < N; ++i)
        cin >> vJewel[i].first >> vJewel[i].second;
    sort(vJewel.begin(), vJewel.end());

    vector<long long> vBack(K);
    for (int i = 0; i < K; ++i)
        cin >> vBack[i];
    sort(vBack.begin(), vBack.end());

    priority_queue<long long> pq;
    int j = 0;
    for (int i = 0; i < K; ++i)
    {
        while (j < N && vJewel[j].first <= vBack[i])
        {
            pq.push(vJewel[j].second);
            ++j;
        }
        if (pq.size())
        {
            iResult+=pq.top();
            pq.pop();
        }
    }

    cout << iResult;

    return 0;
}

우선순위 큐와 정렬 및 그리디 알고리리즘을 이용하여 풀이