[코딩테스트 C++] 보석 도둑

오늘의 문제

https://www.acmicpc.net/status?user_id=huijae0817&problem_id=1202&from_mine=1

보석 도둑

접근 방식

• 처음에 단순하게 음~ 보석비싼것부터 큰가방에 넣을까 하고 풀었다가 틀렸다.
• 예전에 공부했던 Deadline Scheduling알고리즘이 슥 떠오르면서 문제를 풀 수 있었다.
• Deadline Scheduling은 점수가 가장 큰 것을 deadline에 맞춰 배치하는것이다.
• 그러므로 이익(가격)이 큰것을 먼저 배치하되, 가장 딱맞는 위치에(내 무게랑 같거나, 보다 큰 최소의 값) 배치해야하는것
• 그런 역할을 하는 알고리즘이 lower_bound이고 lower_bound를 간단하게 사용할 수 있는 multiset에 가방의 무게를 배치했다.
• 그러면, 가격이 높은순으로 처리하고, 그 높은 가격의 보석의 무게를 lower_bound로 위치를 찾는다. 위치를 찾으면 해당 내용을 삭제, 못찾으면 다음보석으로 넘어간다.

나의 풀이

#include <iostream>
#include <algorithm>
#include <set>
using namespace std;
const int MAX = 300001;
pair<int, int> jew[MAX];
multiset<int> weight;
int n, k;

// 보석 도둑
bool cmp(pair<int, int> a, pair<int, int> b){
    return a.second> b.second;
}
long long solution(){
    long long answer = 0;
    sort(jew, jew + n, cmp);
    for(int i=0;i<n;i++){
        if(weight.lower_bound(jew[i].first) != weight.end()){
            answer += jew[i].second;
            weight.erase(weight.lower_bound(jew[i].first));
            if(weight.empty())
                break;
        }
    }
    return answer;
}

다른 풀이

#include <iostream>
#include <algorithm>
#include <queue>
using namespace std;

int N, K;
long long res = 0;
pair<int, int> gem[300000];
int bag[300000];
priority_queue<int> pq;

int main(){
	ios_base::sync_with_stdio(0);
	cin.tie(0);
	
	cin >> N >> K;
	
	for(int i=0; i<N; i++)
		cin >> gem[i].first >> gem[i].second;
	for(int i=0; i<K; i++)
		cin >> bag[i];
	
	// 보석(무게기준), 가방 오름차순 정렬	
	sort(gem, gem + N);
	sort(bag, bag + K);
	
	// 무게제한에 맞는 보석 일단 모두 넣음
	int cnt = 0;   // 들어간 보석 개수  
	for(int i=0; i<K; i++){
		while(cnt < N && gem[cnt].first <= bag[i])
			pq.push(gem[cnt++].second);
		
		// 비싼거부터 뺀다. 	
		if(!pq.empty()){
			res += pq.top();
			pq.pop();
		}
	}
	
	cout << res;
	
	
	return 0;
}

배울 점

• 와 이렇게 간단하게!
• 무게제한에 맞는 보석을 다 넣고 비싼것부터 뽑는다.
• 나는 보석을 중심으로 문제를 풀었는데, 이사람은 가방을 중심으로 풀었다.