https://www.acmicpc.net/problem/1202
\n문제
세계적인 도둑 상덕이는 보석점을 털기로 결심했다.
\n상덕이가 털 보석점에는 보석이 총 N개 있다. 각 보석은 무게 Mi와 가격 Vi를 가지고 있다. 상덕이는 가방을 K개 가지고 있고, 각 가방에 담을 수 있는 최대 무게는 Ci이다. 가방에는 최대 한 개의 보석만 넣을 수 있다.
\n상덕이가 훔칠 수 있는 보석의 최대 가격을 구하는 프로그램을 작성하시오.
\n입력
첫째 줄에 N과 K가 주어진다. (1 ≤ N, K ≤ 300,000)
\n다음 N개 줄에는 각 보석의 정보 Mi와 Vi가 주어진다. (0 ≤ Mi, Vi ≤ 1,000,000)
\n다음 K개 줄에는 가방에 담을 수 있는 최대 무게 Ci가 주어진다. (1 ≤ Ci ≤ 100,000,000)
\n모든 숫자는 양의 정수이다.
\n출력
첫째 줄에 상덕이가 훔칠 수 있는 보석 가격의 합의 최댓값을 출력한다.
\n처음엔 냅색 문제인 줄 알았는데, 가방에 한번에 하나만 넣어야 한다는 조건이 있었다. 따라서 최적의 답을 찾으려면 가장 크기가 작은 가방에 들어갈 수 있는 보석 중 가치가 가장 큰 보석을 넣어주어야 한다.
\n가방의 들어갈 수 있는 보석은 무게가 최소한 보석의 무게가 된다는 점을 이용하면 모든 가방에 보석을 다 넣어보지 않아도 쉽게 들어갈 보석을 골라낼 수 있다. 우리는 set 컨테이너에 들어있는 lower_bound 를 이용할 것이다.
\nlowerbound 함수는 set에서 인자로 주어진 어떤 값이 찾는데, 만약 그 값이 존재하지 않는다면, 해당 값 보다 큰 값 중 가장 작은 값을 선택해서 그 iterator를 반환한다. set 에 가방들을 모두 넣어두고, lowerbound의 인자로 가장 가치가 큰 보석부터 찾아보면 해당 보석을 담을 수 있는 배낭의 iterator 를 얻게 된다.
\n배낭에 보석을 넣게 되면 더 이상 보석을 넣을 수 없기 때문에, set에서 배낭을 삭제하고 모든 배낭에 보석을 넣을 때까지 위 작업을 반복한다. multiset을 이용한 이유는 일반 set 컨테이너는 중복된 key를 허용하지 않기 때문에 최대 무게가 같은 배낭이 여러개 있을 경우를 처리하지 못하기 때문이다.
\n' '\n#include <cstdio>\n#include <set>\n#include <vector>\n#include <algorithm>\n\nusing namespace std;\n\nvector<pair<int, int>> jewels;\nmultiset<int> bags;\n\nbool comp (pair<int, int> a, pair<int, int> b){\n if (a.second == b.second){\n return a.first < b.first;\n }\n return a.second > b.second;\n}\n\nint main (){\n int N, K;\n scanf(\"%d %d\", &N, &K);\n\n for (int i = 0 ; i < N ; i ++){\n int weight;\n int value;\n\n scanf(\"%d %d\", &weight, &value);\n\n jewels.push_back({weight, value});\n }\n\n sort(jewels.begin(), jewels.end(), comp);\n\n for (int i = 0 ; i < K ; i++){\n int bag;\n scanf(\"%d\", &bag);\n bags.insert(bag);\n }\n\n long long sum = 0;\n for (auto jewel : jewels){\n if (bags.empty()) break;\n auto iter = bags.lower_bound(jewel.first); // 현재 보석의 무게 이상의 첫번째 가방을 찾는다.\n if(iter != bags.end()){\n sum += jewel.second; // 보석의 가치를 더하고\n bags.erase(iter); // 해당 가방은 찾을 대상에서 제외시킨다.\n }\n }\n\n printf(\"%lld\", sum);\n}