[코딩테스트 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;
}
배울 점
- 와 이렇게 간단하게!
- 무게제한에 맞는 보석을 다 넣고 비싼것부터 뽑는다.
- 나는 보석을 중심으로 문제를 풀었는데, 이사람은 가방을 중심으로 풀었다.
Author And Source
이 문제에 관하여([코딩테스트 C++] 보석 도둑), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://velog.io/@huijae0817/코딩테스트-C-보석-도둑저자 귀속: 원작자 정보가 원작자 URL에 포함되어 있으며 저작권은 원작자 소유입니다.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)