BOJ 1202. 보석 도둑

🙇‍♀️[Gold II] 보석 도둑 - 1202

문제 링크

성능 요약

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

분류

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

제출 일자

2024년 3월 12일 12:17:20

문제 설명

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

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

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

입력

첫째 줄에 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)

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

출력

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

🚀풀이

우선순위 큐를 응용한 문제이다.

int n, k;
const int MAX = 300010;
pair<int, int> jewerly[MAX];
int bag[MAX];
priority_queue<int, vector<int>, less<int>> q;

void solve()
{
    cin >> n >> k;
    for (int i = 0; i < n; ++i)
    {
        cin >> jewerly[i].first >> jewerly[i].second;
    }
    for (int i = 0; i < k; ++i)
    {
        cin >> bag[i];
    }

    sort(jewerly, jewerly + n);
    sort(bag, bag + k);

    int idx = 0;
    long long ret = 0;

    for (int i = 0; i < k; ++i)
    {
        // q에 가방의 무게가 된다면 다 집어 넣기
        while (idx < n && bag[i] >= jewerly[idx].first)
        {
            q.push(jewerly[idx].second);
            idx++;
        }
        // 그 중 가장 비싼거를 가방에 넣기
        // 가방 하나 당 하나의 보석이 들어간다.
        if (q.empty() == false)
        {
            ret += q.top();
            q.pop();
        }
    }

    cout << ret;
}

🚀전체 코드

#define _CRT_SECURE_NO_WARNINGS

#include <stdio.h>
#include <string.h>
#include <ctype.h>
#include<iostream>
#include <fstream>
#include <vector>
#include <algorithm>
#include <math.h>
#include <queue>

using namespace std;

int n, k;
const int MAX = 300010;
pair<int, int> jewerly[MAX];
int bag[MAX];
priority_queue<int, vector<int>, less<int>> q;

void solve()
{
    cin >> n >> k;
    for (int i = 0; i < n; ++i)
    {
        cin >> jewerly[i].first >> jewerly[i].second;
    }
    for (int i = 0; i < k; ++i)
    {
        cin >> bag[i];
    }

    sort(jewerly, jewerly + n);
    sort(bag, bag + k);

    int idx = 0;
    long long ret = 0;

    for (int i = 0; i < k; ++i)
    {
        while (idx < n && bag[i] >= jewerly[idx].first)
        {
            q.push(jewerly[idx].second);
            idx++;
        }
        if (q.empty() == false)
        {
            ret += q.top();
            q.pop();
        }
    }

    cout << ret;
}

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    freopen("input.txt", "rt", stdin);

    solve();

    return 0;
}