You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
51 lines
1.4 KiB
51 lines
1.4 KiB
/**
|
|
* File: fractional_knapsack.kt
|
|
* Created Time: 2024-01-25
|
|
* Author: curtishd (1023632660@qq.com)
|
|
*/
|
|
|
|
package chapter_greedy
|
|
|
|
/* 物品 */
|
|
class Item(
|
|
val w: Int, // 物品
|
|
val v: Int // 物品价值
|
|
)
|
|
|
|
/* 分数背包：贪心 */
|
|
fun fractionalKnapsack(wgt: IntArray, _val: IntArray, c: Int): Double {
|
|
// 创建物品列表，包含两个属性：重量、价值
|
|
var cap = c
|
|
val items = arrayOfNulls<Item>(wgt.size)
|
|
for (i in wgt.indices) {
|
|
items[i] = Item(wgt[i], _val[i])
|
|
}
|
|
// 按照单位价值 item.v / item.w 从高到低进行排序
|
|
items.sortBy { item: Item? -> -(item!!.v.toDouble() / item.w) }
|
|
// 循环贪心选择
|
|
var res = 0.0
|
|
for (item in items) {
|
|
if (item!!.w <= cap) {
|
|
// 若剩余容量充足，则将当前物品整个装进背包
|
|
res += item.v
|
|
cap -= item.w
|
|
} else {
|
|
// 若剩余容量不足，则将当前物品的一部分装进背包
|
|
res += item.v.toDouble() / item.w * cap
|
|
// 已无剩余容量，因此跳出循环
|
|
break
|
|
}
|
|
}
|
|
return res
|
|
}
|
|
|
|
/* Driver Code */
|
|
fun main() {
|
|
val wgt = intArrayOf(10, 20, 30, 40, 50)
|
|
val _val = intArrayOf(50, 120, 150, 210, 240)
|
|
val cap = 50
|
|
|
|
// 贪心算法
|
|
val res = fractionalKnapsack(wgt, _val, cap)
|
|
println("不超过背包容量的最大物品价值为 $res")
|
|
} |