728x90
문제
세계적인 도둑 상덕이는 보석점을 털기로 결심했다.
상덕이가 털 보석점에는 보석이 총 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)
모든 숫자는 양의 정수이다.
출력
첫째 줄에 상덕이가 훔칠 수 있는 보석 가격의 합의 최댓값을 출력한다.
예제 입력 1 복사
2 1 5 10 100 100 11
예제 출력 1 복사
10
예제 입력 2 복사
3 2 1 65 5 23 2 99 10 2
예제 출력 2 복사
164
package BaekJoon.Greedy;
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.Arrays;
import java.util.Collections;
import java.util.Comparator;
import java.util.PriorityQueue;
import java.util.StringTokenizer;
public class 보석도둑 {
static int n,k;
static class Pair{
int x,y;
Pair(int x, int y){
this.x =x;
this.y =y;
}
}
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
StringTokenizer st = new StringTokenizer(br.readLine());
n = Integer.parseInt(st.nextToken());
k = Integer.parseInt(st.nextToken());
Pair[] arr = new Pair[n];
int[] bags = new int[k];
for(int i=0; i<n; i++) {
st = new StringTokenizer(br.readLine());
int m = Integer.parseInt(st.nextToken());
int v = Integer.parseInt(st.nextToken());
arr[i] = new Pair(m,v);
}
for(int i=0; i<k; i++) {
bags[i] = Integer.parseInt(br.readLine());
}
Arrays.sort(arr, new Comparator<Pair>() {
@Override
public int compare(Pair a, Pair b) {
if(a.x==b.x) {return b.y-a.y;}
return a.x-b.x;
}
});
Arrays.sort(bags);
long totalv=0; // 총 가격이 int 를 벗어날 수 있으므로 long 으로 해준다.
PriorityQueue<Integer>pq =new PriorityQueue<>(Comparator.reverseOrder());
for(int i=0 ,j=0; i<k; i++) {
while(j<arr.length && arr[j].x <=bags[i]) {
pq.offer(arr[j].y);
j++;
}
if(!pq.isEmpty()) {
totalv+=pq.poll();
}
}
bw.write(totalv + "\n");
bw.flush();
bw.close();
br.close();
}
}- 무게가 작은것들을 모두 힙에 담고 그중 가격이 가장 높은 것을 고른다 -> nlogn 시간복잡도
- loop문 2개 해서 n^2의 시간복잡도 발생보다 낫다.