1202번: 보석 도둑

문제 출처 : https://www.acmicpc.net/problem/1202

1202번: 보석 도둑

언어 : Kotlin

 

문제 설명 :

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

상덕이가 털 보석점에는 보석이 총 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.5초
• 메모리 제한 : 256MB

 

입출력 예 :

입력	출력
2 1 5 10 100 100 11	10
3 2 1 65 5 23 2 99 10 2	164

힌트 :

• 두 번째 예제의 경우 첫 번째 보석을 두 번째 가방에, 세 번째 보석을 첫 번째 가방에 넣으면 된다.

 

풀이 : 

import java.io.BufferedWriter
import java.io.OutputStreamWriter
import java.util.*

data class Jewel(val weight: Int, val value: Int)

fun main() = with(System.`in`.bufferedReader()) {
    val bw = BufferedWriter(OutputStreamWriter(System.out))

    val (n, k) = readLine().split(" ").map { it.toInt() }
    val jewelList = ArrayList<Jewel>()
    val backpack = ArrayList<Int>()
    val pq = PriorityQueue(Comparator.reverseOrder<Int>())
    var sum: Long = 0
    var pos = 0
    repeat(n) {
        val (m, v) = readLine().split(" ").map { it.toInt() }
        jewelList += Jewel(m, v)
    }
    repeat(k) { backpack += readLine().toInt() }

    jewelList.sortWith { j1, j2 -> if (j1.weight == j2.weight) j2.value - j1.value else j1.weight - j2.weight }
    backpack.sort()

    for (it in backpack) {
        for (i in pos until jewelList.size) {
            if (jewelList[i].weight <= it) { pq.offer(jewelList[i].value); pos = i + 1 } else break
        }

        if (!pq.isEmpty()) sum += pq.poll()
    }

    bw.write("$sum")
    bw.flush()
    bw.close()
}