mirror of
https://github.com/krahets/hello-algo.git
synced 2024-12-27 00:56:29 +08:00
c1adeb2399
* feat(go/dp): support climbing stairs * feat(go/dp): support knapsack * feat(go/dp): coin_change & edit_distance
54 lines
1.2 KiB
Go
54 lines
1.2 KiB
Go
// File: coin_change_ii.go
|
||
// Created Time: 2023-07-23
|
||
// Author: Reanon (793584285@qq.com)
|
||
|
||
package chapter_dynamic_programming
|
||
|
||
/* 零钱兑换 II:动态规划 */
|
||
func coinChangeIIDP(coins []int, amt int) int {
|
||
n := len(coins)
|
||
// 初始化 dp 表
|
||
dp := make([][]int, n+1)
|
||
for i := 0; i <= n; i++ {
|
||
dp[i] = make([]int, amt+1)
|
||
}
|
||
// 初始化首列
|
||
for i := 0; i <= n; i++ {
|
||
dp[i][0] = 1
|
||
}
|
||
// 状态转移:其余行列
|
||
for i := 1; i <= n; i++ {
|
||
for a := 1; a <= amt; a++ {
|
||
if coins[i-1] > a {
|
||
// 若超过背包容量,则不选硬币 i
|
||
dp[i][a] = dp[i-1][a]
|
||
} else {
|
||
// 不选和选硬币 i 这两种方案的较小值
|
||
dp[i][a] = dp[i-1][a] + dp[i][a-coins[i-1]]
|
||
}
|
||
}
|
||
}
|
||
return dp[n][amt]
|
||
}
|
||
|
||
/* 零钱兑换 II:状态压缩后的动态规划 */
|
||
func coinChangeIIDPComp(coins []int, amt int) int {
|
||
n := len(coins)
|
||
// 初始化 dp 表
|
||
dp := make([]int, amt+1)
|
||
dp[0] = 1
|
||
// 状态转移
|
||
for i := 1; i <= n; i++ {
|
||
// 倒序遍历
|
||
for a := 1; a <= amt; a++ {
|
||
if coins[i-1] > a {
|
||
// 若超过背包容量,则不选硬币 i
|
||
dp[a] = dp[a]
|
||
} else {
|
||
// 不选和选硬币 i 这两种方案之和
|
||
dp[a] = dp[a] + dp[a-coins[i-1]]
|
||
}
|
||
}
|
||
}
|
||
return dp[amt]
|
||
}
|