[백준 1202] 보석 도둑 (JAVA)

https://www.acmicpc.net/problem/1202

1202번: 보석 도둑

 

내림차순 정렬한 우선순위큐를 이용하여 해결할 수 있었다.

객체를 하나 만들어서 보석의 무게와 가격 속성을 갖는다. 무게순으로 오름차순 정렬한다.

또한 가방도 오름차순 정렬한다.

가방에 넣을 수 있는 보석이라면 큐에 넣고 q.poll()을 하여 result값에 더해준다.

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.Collections;
import java.util.Comparator;
import java.util.PriorityQueue;
import java.util.StringTokenizer;

public class Main {
	static int n, k;
	static int[][] narr;
	static int[] karr;

	static class Jewel {
		int g;
		int price;

		Jewel(int g, int price) {
			this.g = g;
			this.price = price;
		}

		int getG() {
			return g;
		}

		int getPrice() {
			return price;
		}
	}

	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());
		n = Integer.parseInt(st.nextToken());
		k = Integer.parseInt(st.nextToken());

		Jewel[] jewel = new Jewel[n];
		karr = new int[k];
		for (int i = 0; i < n; i++) {
			st = new StringTokenizer(br.readLine());
			int g = Integer.parseInt(st.nextToken());
			int price = Integer.parseInt(st.nextToken());
			jewel[i] = new Jewel(g, price);
		}
		for (int i = 0; i < k; i++) {
			karr[i] = Integer.parseInt(br.readLine());
		}

		Arrays.sort(jewel, new Comparator<Jewel>() {
			public int compare(Jewel j1, Jewel j2) {
				return j1.getG() - j2.getG();
			}
		});

		Arrays.sort(karr);

		PriorityQueue<Integer> pq = new PriorityQueue(Collections.reverseOrder());

		long result = 0;
		int idx = 0;
		for (int i = 0; i < k; i++) {
			while (idx < n && jewel[idx].g <= karr[i]) { //가방은 무게 오름차순으로 되어 있기에 i=0번째에 들어가는 거는 i=1에도 가능..
				pq.add(jewel[idx++].price);
			}

			if (!pq.isEmpty())
				result += pq.poll();
		}

		System.out.println(result);
	}

}