| 시간 제한 | 메모리 제한 | 제출 | 정답 | 맞은 사람 | 정답 비율 |
|---|---|---|---|---|---|
| 1 초 | 256 MB | 14419 | 3356 | 2365 | 23.082% |
문제
세계적인 도둑 상덕이는 보석점을 털기로 결심했다.
상덕이가 털 보석점에는 보석이 총 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)
모든 숫자는 양의 정수이다.
출력
첫째 줄에 상덕이가 훔칠 수 있는 보석 가격의 합의 최댓값을 출력한다.
접근
전형적인 그리디 알고리즘 문제이다.
각 가방을 순회하면서 그때그때 최고로 높은 가치의 보석을 넣어주면 된다.
풀이
투 포인터 형식으로 보석의 idx를 만들었다.
이 후 가방을 순회할 때, 넣을 수 있는 보석들을 pq에 모두 넣어주었다.
pq가 비지 않았으면, top을 정답에 더해주면 된다.
#include <bits/stdc++.h>
using namespace std;
#define ll long long int
#define FUP(i, a, b) for(int i = a; i <= b; i++)
#define FDOWN(i, a, b) for(int i = a; i >= b; i--)
#define MS(a, b) memset(a, b, sizeof(a))
#define ALL(v) v.begin(), v.end()
#define CIN(a) cin >> a;
#define CIN2(a, b) cin >> a >> b
#define CIN3(a, b, c) cin >> a >> b >> c
#define COUT(a) cout << a
#define COUT2(a, b) cout << a << ' ' << b
#define COUT3(a, b, c) cout << a << ' ' << b << ' ' << c
#define ENDL cout << '\n'
int dy[4] = { -1, 1, 0, 0 };
int dx[4] = { 0, 0, 1, -1 };
int N, M, K, bag[300000];
ll ans = 0;
pair<int, int> jewelry[300000];
int main()
{
ios_base::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
CIN2(N, K);
FUP(i, 0, N - 1) CIN2(jewelry[i].first, jewelry[i].second);
FUP(i, 0, K - 1) CIN(bag[i]);
sort(jewelry, jewelry + N);
sort(bag, bag + K);
int idx = 0;
priority_queue<int> pq;
FUP(i, 0, K - 1)
{
while (idx < N && jewelry[idx].first <= bag[i])
{
pq.push(jewelry[idx++].second);
}
if (!pq.empty())
{
ans += pq.top();
pq.pop();
}
}
COUT(ans);
return 0;
}
Beginner