Add kotlin codes for the chapter of greedy (#1103)
* feat(kotlin): add kotlin code for dynamic programming. * Update knapsack.kt * feat(kotlin): add kotlin codes for graph. * style(kotlin): reformatted the codes. * feat(kotlin): add kotlin codes for the chapter of greedy. * Update max_product_cutting.ktpull/1105/head^2
parent
a032d68c7c
commit
89679a4a13
@ -0,0 +1,53 @@
|
|||||||
|
/**
|
||||||
|
* File: coin_change_greedy.kt
|
||||||
|
* Created Time: 2024-01-25
|
||||||
|
* Author: curtishd (1023632660@qq.com)
|
||||||
|
*/
|
||||||
|
|
||||||
|
package chapter_greedy
|
||||||
|
|
||||||
|
/* 零钱兑换:贪心 */
|
||||||
|
fun coinChangeGreedy(coins: IntArray, amt: Int): Int {
|
||||||
|
// 假设 coins 列表有序
|
||||||
|
var am = amt
|
||||||
|
var i = coins.size - 1
|
||||||
|
var count = 0
|
||||||
|
// 循环进行贪心选择,直到无剩余金额
|
||||||
|
while (am > 0) {
|
||||||
|
// 找到小于且最接近剩余金额的硬币
|
||||||
|
while (i > 0 && coins[i] > am) {
|
||||||
|
i--
|
||||||
|
}
|
||||||
|
// 选择 coins[i]
|
||||||
|
am -= coins[i]
|
||||||
|
count++
|
||||||
|
}
|
||||||
|
// 若未找到可行方案,则返回 -1
|
||||||
|
return if (am == 0) count else -1
|
||||||
|
}
|
||||||
|
|
||||||
|
/* Driver Code */
|
||||||
|
fun main() {
|
||||||
|
// 贪心:能够保证找到全局最优解
|
||||||
|
var coins = intArrayOf(1, 5, 10, 20, 50, 100)
|
||||||
|
var amt = 186
|
||||||
|
var res = coinChangeGreedy(coins, amt)
|
||||||
|
println("\ncoins = ${coins.contentToString()}, amt = $amt")
|
||||||
|
println("凑到 $amt 所需的最少硬币数量为 $res")
|
||||||
|
|
||||||
|
// 贪心:无法保证找到全局最优解
|
||||||
|
coins = intArrayOf(1, 20, 50)
|
||||||
|
amt = 60
|
||||||
|
res = coinChangeGreedy(coins, amt)
|
||||||
|
println("\ncoins = ${coins.contentToString()}, amt = $amt")
|
||||||
|
println("凑到 $amt 所需的最少硬币数量为 $res")
|
||||||
|
println("实际上需要的最少数量为 3 ,即 20 + 20 + 20")
|
||||||
|
|
||||||
|
// 贪心:无法保证找到全局最优解
|
||||||
|
coins = intArrayOf(1, 49, 50)
|
||||||
|
amt = 98
|
||||||
|
res = coinChangeGreedy(coins, amt)
|
||||||
|
println("\ncoins = ${coins.contentToString()}, amt = $amt")
|
||||||
|
println("凑到 $amt 所需的最少硬币数量为 $res")
|
||||||
|
println("实际上需要的最少数量为 2 ,即 49 + 49")
|
||||||
|
}
|
@ -0,0 +1,57 @@
|
|||||||
|
/**
|
||||||
|
* File: fractional_knapsack.kt
|
||||||
|
* Created Time: 2024-01-25
|
||||||
|
* Author: curtishd (1023632660@qq.com)
|
||||||
|
*/
|
||||||
|
|
||||||
|
package chapter_greedy
|
||||||
|
|
||||||
|
import java.util.*
|
||||||
|
|
||||||
|
/* 物品 */
|
||||||
|
class Item(
|
||||||
|
val w: Int, // 物品
|
||||||
|
val v: Int // 物品价值
|
||||||
|
)
|
||||||
|
|
||||||
|
/* 分数背包:贪心 */
|
||||||
|
fun fractionalKnapsack(
|
||||||
|
wgt: IntArray,
|
||||||
|
value: IntArray,
|
||||||
|
c: Int
|
||||||
|
): Double {
|
||||||
|
// 创建物品列表,包含两个属性:重量、价值
|
||||||
|
var cap = c
|
||||||
|
val items = arrayOfNulls<Item>(wgt.size)
|
||||||
|
for (i in wgt.indices) {
|
||||||
|
items[i] = Item(wgt[i], value[i])
|
||||||
|
}
|
||||||
|
// 按照单位价值 item.v / item.w 从高到低进行排序
|
||||||
|
Arrays.sort(items, Comparator.comparingDouble { item: Item -> -(item.v.toDouble() / item.w) })
|
||||||
|
// 循环贪心选择
|
||||||
|
var res = 0.0
|
||||||
|
for (item in items) {
|
||||||
|
if (item!!.w <= cap) {
|
||||||
|
// 若剩余容量充足,则将当前物品整个装进背包
|
||||||
|
res += item.v.toDouble()
|
||||||
|
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 values = intArrayOf(50, 120, 150, 210, 240)
|
||||||
|
val cap = 50
|
||||||
|
|
||||||
|
// 贪心算法
|
||||||
|
val res = fractionalKnapsack(wgt, values, cap)
|
||||||
|
println("不超过背包容量的最大物品价值为 $res")
|
||||||
|
}
|
Loading…
Reference in new issue