[백준] 1202번 : 보석 도둑 Gold2(골드2) - JAVA[자바]

[Gold II] 보석 도둑 - 1202

문제 링크

성능 요약

메모리: 326064 KB, 시간: 2880 ms

분류

자료 구조, 그리디 알고리즘, 우선순위 큐, 정렬

제출 일자

2023년 10월 30일 14:33:36

문제 설명

세계적인 도둑 상덕이는 보석점을 털기로 결심했다.

상덕이가 털 보석점에는 보석이 총 N개 있다. 각 보석은 무게 M_i와 가격 V_i를 가지고 있다. 상덕이는 가방을 K개 가지고 있고, 각 가방에 담을 수 있는 최대 무게는 C_i이다. 가방에는 최대 한 개의 보석만 넣을 수 있다.

상덕이가 훔칠 수 있는 보석의 최대 가격을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N과 K가 주어진다. (1 ≤ N, K ≤ 300,000)

다음 N개 줄에는 각 보석의 정보 M_i와 V_i가 주어진다. (0 ≤ M_i, V_i ≤ 1,000,000)

다음 K개 줄에는 가방에 담을 수 있는 최대 무게 C_i가 주어진다. (1 ≤ C_i ≤ 100,000,000)

모든 숫자는 양의 정수이다.

출력

첫째 줄에 상덕이가 훔칠 수 있는 보석 가격의 합의 최댓값을 출력한다.

import java.util.Arrays;
import java.util.PriorityQueue;
import java.util.Scanner;

class Main{
    static class Jewel{
        int weight;
        int value;

        public Jewel(int weight, int value) {
            this.weight = weight;
            this.value = value;
        }
    }
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int k = sc.nextInt();
        long answer = 0;
        Jewel[] jewels = new Jewel[n];
        int bags[] = new int[k];
        for(int i = 0; i < n; i++){
            jewels[i] = new Jewel(sc.nextInt(), sc.nextInt());
        }
        Arrays.sort(jewels,(o1,o2)->{
            Jewel j1 = (Jewel) o1;
            Jewel j2 = (Jewel) o2;
            if(j1.weight == j2.weight){
                return j2.value - j1.value;
            }
            return j1.weight - j2.weight;
        });
        for(int i = 0; i < k; i++){
            bags[i] = sc.nextInt();
        }
        Arrays.sort(bags);
        PriorityQueue<Integer> pq = new PriorityQueue<>((o1,o2) -> o2-o1);
        for(int i = 0,j =0; i < k; i++){
            while(j < n && jewels[j].weight <= bags[i]){
                pq.offer(jewels[j++].value);
            }
            if(!pq.isEmpty()){
                answer += pq.poll();
            }
        }
        System.out.println(answer);
    }

}