mirror of
https://github.com/krahets/hello-algo.git
synced 2026-07-11 23:16:07 +00:00
Re-translate the Japanese version (#1871)
* Retranslate Japanese docs with GPT-5.4 * Retranslate Japanese code with GPT-5.4
This commit is contained in:
@@ -0,0 +1,45 @@
|
||||
/**
|
||||
* File: climbing_stairs_backtrack.kt
|
||||
* Created Time: 2024-01-25
|
||||
* Author: curtishd (1023632660@qq.com)
|
||||
*/
|
||||
|
||||
package chapter_dynamic_programming
|
||||
|
||||
/* バックトラッキング */
|
||||
fun backtrack(
|
||||
choices: MutableList<Int>,
|
||||
state: Int,
|
||||
n: Int,
|
||||
res: MutableList<Int>
|
||||
) {
|
||||
// 第 n 段に到達したら、方法数を 1 増やす
|
||||
if (state == n)
|
||||
res[0] = res[0] + 1
|
||||
// すべての選択肢を走査
|
||||
for (choice in choices) {
|
||||
// 枝刈り: 第 n 段を超えないようにする
|
||||
if (state + choice > n) continue
|
||||
// 試行: 選択を行い、状態を更新
|
||||
backtrack(choices, state + choice, n, res)
|
||||
// バックトラック
|
||||
}
|
||||
}
|
||||
|
||||
/* 階段登り:バックトラッキング */
|
||||
fun climbingStairsBacktrack(n: Int): Int {
|
||||
val choices = mutableListOf(1, 2) // 1 段または 2 段上ることを選べる
|
||||
val state = 0 // 第 0 段から上り始める
|
||||
val res = mutableListOf<Int>()
|
||||
res.add(0) // res[0] を使って方法数を記録する
|
||||
backtrack(choices, state, n, res)
|
||||
return res[0]
|
||||
}
|
||||
|
||||
/* Driver Code */
|
||||
fun main() {
|
||||
val n = 9
|
||||
|
||||
val res = climbingStairsBacktrack(n)
|
||||
println("$n 段の階段の登り方は全部で $res 通り")
|
||||
}
|
||||
@@ -0,0 +1,35 @@
|
||||
/**
|
||||
* File: climbing_stairs_constraint_dp.kt
|
||||
* Created Time: 2024-01-25
|
||||
* Author: curtishd (1023632660@qq.com)
|
||||
*/
|
||||
|
||||
package chapter_dynamic_programming
|
||||
|
||||
/* 制約付き階段登り:動的計画法 */
|
||||
fun climbingStairsConstraintDP(n: Int): Int {
|
||||
if (n == 1 || n == 2) {
|
||||
return 1
|
||||
}
|
||||
// 部分問題の解を保存するために dp テーブルを初期化
|
||||
val dp = Array(n + 1) { IntArray(3) }
|
||||
// 初期状態:最小部分問題の解をあらかじめ設定
|
||||
dp[1][1] = 1
|
||||
dp[1][2] = 0
|
||||
dp[2][1] = 0
|
||||
dp[2][2] = 1
|
||||
// 状態遷移:小さい部分問題から大きい部分問題へ順に解く
|
||||
for (i in 3..n) {
|
||||
dp[i][1] = dp[i - 1][2]
|
||||
dp[i][2] = dp[i - 2][1] + dp[i - 2][2]
|
||||
}
|
||||
return dp[n][1] + dp[n][2]
|
||||
}
|
||||
|
||||
/* Driver Code */
|
||||
fun main() {
|
||||
val n = 9
|
||||
|
||||
val res = climbingStairsConstraintDP(n)
|
||||
println("$n 段の階段の登り方は全部で $res 通り")
|
||||
}
|
||||
@@ -0,0 +1,29 @@
|
||||
/**
|
||||
* File: climbing_stairs_dfs.kt
|
||||
* Created Time: 2024-01-25
|
||||
* Author: curtishd (1023632660@qq.com)
|
||||
*/
|
||||
|
||||
package chapter_dynamic_programming
|
||||
|
||||
/* 検索 */
|
||||
fun dfs(i: Int): Int {
|
||||
// dp[1] と dp[2] は既知なので返す
|
||||
if (i == 1 || i == 2) return i
|
||||
// dp[i] = dp[i-1] + dp[i-2]
|
||||
val count = dfs(i - 1) + dfs(i - 2)
|
||||
return count
|
||||
}
|
||||
|
||||
/* 階段登り:探索 */
|
||||
fun climbingStairsDFS(n: Int): Int {
|
||||
return dfs(n)
|
||||
}
|
||||
|
||||
/* Driver Code */
|
||||
fun main() {
|
||||
val n = 9
|
||||
|
||||
val res = climbingStairsDFS(n)
|
||||
println("$n 段の階段の登り方は全部で $res 通り")
|
||||
}
|
||||
@@ -0,0 +1,36 @@
|
||||
/**
|
||||
* File: climbing_stairs_dfs_mem.kt
|
||||
* Created Time: 2024-01-25
|
||||
* Author: curtishd (1023632660@qq.com)
|
||||
*/
|
||||
|
||||
package chapter_dynamic_programming
|
||||
|
||||
/* メモ化探索 */
|
||||
fun dfs(i: Int, mem: IntArray): Int {
|
||||
// dp[1] と dp[2] は既知なので返す
|
||||
if (i == 1 || i == 2) return i
|
||||
// dp[i] の記録があれば、それをそのまま返す
|
||||
if (mem[i] != -1) return mem[i]
|
||||
// dp[i] = dp[i-1] + dp[i-2]
|
||||
val count = dfs(i - 1, mem) + dfs(i - 2, mem)
|
||||
// dp[i] を記録する
|
||||
mem[i] = count
|
||||
return count
|
||||
}
|
||||
|
||||
/* 階段登り:メモ化探索 */
|
||||
fun climbingStairsDFSMem(n: Int): Int {
|
||||
// mem[i] は第 i 段まで上る方法の総数を記録し、-1 は未記録を表す
|
||||
val mem = IntArray(n + 1)
|
||||
mem.fill(-1)
|
||||
return dfs(n, mem)
|
||||
}
|
||||
|
||||
/* Driver Code */
|
||||
fun main() {
|
||||
val n = 9
|
||||
|
||||
val res = climbingStairsDFSMem(n)
|
||||
println("$n 段の階段の登り方は全部で $res 通り")
|
||||
}
|
||||
@@ -0,0 +1,46 @@
|
||||
/**
|
||||
* File: climbing_stairs_dp.kt
|
||||
* Created Time: 2024-01-25
|
||||
* Author: curtishd (1023632660@qq.com)
|
||||
*/
|
||||
|
||||
package chapter_dynamic_programming
|
||||
|
||||
/* 階段登り:動的計画法 */
|
||||
fun climbingStairsDP(n: Int): Int {
|
||||
if (n == 1 || n == 2) return n
|
||||
// 部分問題の解を保存するために dp テーブルを初期化
|
||||
val dp = IntArray(n + 1)
|
||||
// 初期状態:最小部分問題の解をあらかじめ設定
|
||||
dp[1] = 1
|
||||
dp[2] = 2
|
||||
// 状態遷移:小さい部分問題から大きい部分問題へ順に解く
|
||||
for (i in 3..n) {
|
||||
dp[i] = dp[i - 1] + dp[i - 2]
|
||||
}
|
||||
return dp[n]
|
||||
}
|
||||
|
||||
/* 階段登り:空間最適化した動的計画法 */
|
||||
fun climbingStairsDPComp(n: Int): Int {
|
||||
if (n == 1 || n == 2) return n
|
||||
var a = 1
|
||||
var b = 2
|
||||
for (i in 3..n) {
|
||||
val temp = b
|
||||
b += a
|
||||
a = temp
|
||||
}
|
||||
return b
|
||||
}
|
||||
|
||||
/* Driver Code */
|
||||
fun main() {
|
||||
val n = 9
|
||||
|
||||
var res = climbingStairsDP(n)
|
||||
println("$n 段の階段の登り方は全部で $res 通り")
|
||||
|
||||
res = climbingStairsDPComp(n)
|
||||
println("$n 段の階段の登り方は全部で $res 通り")
|
||||
}
|
||||
@@ -0,0 +1,71 @@
|
||||
/**
|
||||
* File: coin_change.kt
|
||||
* Created Time: 2024-01-25
|
||||
* Author: curtishd (1023632660@qq.com)
|
||||
*/
|
||||
|
||||
package chapter_dynamic_programming
|
||||
|
||||
import kotlin.math.min
|
||||
|
||||
/* コイン両替:動的計画法 */
|
||||
fun coinChangeDP(coins: IntArray, amt: Int): Int {
|
||||
val n = coins.size
|
||||
val MAX = amt + 1
|
||||
// dp テーブルを初期化
|
||||
val dp = Array(n + 1) { IntArray(amt + 1) }
|
||||
// 状態遷移:先頭行と先頭列
|
||||
for (a in 1..amt) {
|
||||
dp[0][a] = MAX
|
||||
}
|
||||
// 状態遷移: 残りの行と列
|
||||
for (i in 1..n) {
|
||||
for (a in 1..amt) {
|
||||
if (coins[i - 1] > a) {
|
||||
// 目標金額を超えるなら硬貨 i は選ばない
|
||||
dp[i][a] = dp[i - 1][a]
|
||||
} else {
|
||||
// 硬貨 i を選ばない場合と選ぶ場合の小さい方
|
||||
dp[i][a] = min(dp[i - 1][a], dp[i][a - coins[i - 1]] + 1)
|
||||
}
|
||||
}
|
||||
}
|
||||
return if (dp[n][amt] != MAX) dp[n][amt] else -1
|
||||
}
|
||||
|
||||
/* コイン交換:空間最適化後の動的計画法 */
|
||||
fun coinChangeDPComp(coins: IntArray, amt: Int): Int {
|
||||
val n = coins.size
|
||||
val MAX = amt + 1
|
||||
// dp テーブルを初期化
|
||||
val dp = IntArray(amt + 1)
|
||||
dp.fill(MAX)
|
||||
dp[0] = 0
|
||||
// 状態遷移
|
||||
for (i in 1..n) {
|
||||
for (a in 1..amt) {
|
||||
if (coins[i - 1] > a) {
|
||||
// 目標金額を超えるなら硬貨 i は選ばない
|
||||
dp[a] = dp[a]
|
||||
} else {
|
||||
// 硬貨 i を選ばない場合と選ぶ場合の小さい方
|
||||
dp[a] = min(dp[a], dp[a - coins[i - 1]] + 1)
|
||||
}
|
||||
}
|
||||
}
|
||||
return if (dp[amt] != MAX) dp[amt] else -1
|
||||
}
|
||||
|
||||
/* Driver Code */
|
||||
fun main() {
|
||||
val coins = intArrayOf(1, 2, 5)
|
||||
val amt = 4
|
||||
|
||||
// 動的計画法
|
||||
var res = coinChangeDP(coins, amt)
|
||||
println("目標金額を作るのに必要な最小硬貨枚数は $res")
|
||||
|
||||
// 空間最適化後の動的計画法
|
||||
res = coinChangeDPComp(coins, amt)
|
||||
println("目標金額を作るのに必要な最小硬貨枚数は $res")
|
||||
}
|
||||
@@ -0,0 +1,66 @@
|
||||
/**
|
||||
* File: coin_change_ii.kt
|
||||
* Created Time: 2024-01-25
|
||||
* Author: curtishd (1023632660@qq.com)
|
||||
*/
|
||||
|
||||
package chapter_dynamic_programming
|
||||
|
||||
/* コイン両替 II:動的計画法 */
|
||||
fun coinChangeIIDP(coins: IntArray, amt: Int): Int {
|
||||
val n = coins.size
|
||||
// dp テーブルを初期化
|
||||
val dp = Array(n + 1) { IntArray(amt + 1) }
|
||||
// 先頭列を初期化する
|
||||
for (i in 0..n) {
|
||||
dp[i][0] = 1
|
||||
}
|
||||
// 状態遷移
|
||||
for (i in 1..n) {
|
||||
for (a in 1..amt) {
|
||||
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:空間最適化した動的計画法 */
|
||||
fun coinChangeIIDPComp(coins: IntArray, amt: Int): Int {
|
||||
val n = coins.size
|
||||
// dp テーブルを初期化
|
||||
val dp = IntArray(amt + 1)
|
||||
dp[0] = 1
|
||||
// 状態遷移
|
||||
for (i in 1..n) {
|
||||
for (a in 1..amt) {
|
||||
if (coins[i - 1] > a) {
|
||||
// 目標金額を超えるなら硬貨 i は選ばない
|
||||
dp[a] = dp[a]
|
||||
} else {
|
||||
// コイン i を選ばない場合と選ぶ場合の和
|
||||
dp[a] = dp[a] + dp[a - coins[i - 1]]
|
||||
}
|
||||
}
|
||||
}
|
||||
return dp[amt]
|
||||
}
|
||||
|
||||
/* Driver Code */
|
||||
fun main() {
|
||||
val coins = intArrayOf(1, 2, 5)
|
||||
val amt = 5
|
||||
|
||||
// 動的計画法
|
||||
var res = coinChangeIIDP(coins, amt)
|
||||
println("目標金額を作る硬貨の組み合わせ数は $res")
|
||||
|
||||
// 空間最適化後の動的計画法
|
||||
res = coinChangeIIDPComp(coins, amt)
|
||||
println("目標金額を作る硬貨の組み合わせ数は $res")
|
||||
}
|
||||
@@ -0,0 +1,143 @@
|
||||
/**
|
||||
* File: edit_distance.kt
|
||||
* Created Time: 2024-01-25
|
||||
* Author: curtishd (1023632660@qq.com)
|
||||
*/
|
||||
|
||||
package chapter_dynamic_programming
|
||||
|
||||
import kotlin.math.min
|
||||
|
||||
/* 編集距離:総当たり探索 */
|
||||
fun editDistanceDFS(
|
||||
s: String,
|
||||
t: String,
|
||||
i: Int,
|
||||
j: Int
|
||||
): Int {
|
||||
// s と t がともに空なら 0 を返す
|
||||
if (i == 0 && j == 0) return 0
|
||||
// s が空なら t の長さを返す
|
||||
if (i == 0) return j
|
||||
// t が空なら s の長さを返す
|
||||
if (j == 0) return i
|
||||
// 2 つの文字が等しければ、その 2 文字をそのままスキップする
|
||||
if (s[i - 1] == t[j - 1]) return editDistanceDFS(s, t, i - 1, j - 1)
|
||||
// 最小編集回数 = 挿入・削除・置換の 3 操作における最小編集回数 + 1
|
||||
val insert = editDistanceDFS(s, t, i, j - 1)
|
||||
val delete = editDistanceDFS(s, t, i - 1, j)
|
||||
val replace = editDistanceDFS(s, t, i - 1, j - 1)
|
||||
// 最小編集回数を返す
|
||||
return min(min(insert, delete), replace) + 1
|
||||
}
|
||||
|
||||
/* 編集距離:メモ化探索 */
|
||||
fun editDistanceDFSMem(
|
||||
s: String,
|
||||
t: String,
|
||||
mem: Array<IntArray>,
|
||||
i: Int,
|
||||
j: Int
|
||||
): Int {
|
||||
// s と t がともに空なら 0 を返す
|
||||
if (i == 0 && j == 0) return 0
|
||||
// s が空なら t の長さを返す
|
||||
if (i == 0) return j
|
||||
// t が空なら s の長さを返す
|
||||
if (j == 0) return i
|
||||
// 記録済みなら、それをそのまま返す
|
||||
if (mem[i][j] != -1) return mem[i][j]
|
||||
// 2 つの文字が等しければ、その 2 文字をそのままスキップする
|
||||
if (s[i - 1] == t[j - 1]) return editDistanceDFSMem(s, t, mem, i - 1, j - 1)
|
||||
// 最小編集回数 = 挿入・削除・置換の 3 操作における最小編集回数 + 1
|
||||
val insert = editDistanceDFSMem(s, t, mem, i, j - 1)
|
||||
val delete = editDistanceDFSMem(s, t, mem, i - 1, j)
|
||||
val replace = editDistanceDFSMem(s, t, mem, i - 1, j - 1)
|
||||
// 最小編集回数を記録して返す
|
||||
mem[i][j] = min(min(insert, delete), replace) + 1
|
||||
return mem[i][j]
|
||||
}
|
||||
|
||||
/* 編集距離:動的計画法 */
|
||||
fun editDistanceDP(s: String, t: String): Int {
|
||||
val n = s.length
|
||||
val m = t.length
|
||||
val dp = Array(n + 1) { IntArray(m + 1) }
|
||||
// 状態遷移:先頭行と先頭列
|
||||
for (i in 1..n) {
|
||||
dp[i][0] = i
|
||||
}
|
||||
for (j in 1..m) {
|
||||
dp[0][j] = j
|
||||
}
|
||||
// 状態遷移: 残りの行と列
|
||||
for (i in 1..n) {
|
||||
for (j in 1..m) {
|
||||
if (s[i - 1] == t[j - 1]) {
|
||||
// 2 つの文字が等しければ、その 2 文字をそのままスキップする
|
||||
dp[i][j] = dp[i - 1][j - 1]
|
||||
} else {
|
||||
// 最小編集回数 = 挿入・削除・置換の 3 操作における最小編集回数 + 1
|
||||
dp[i][j] = min(min(dp[i][j - 1], dp[i - 1][j]), dp[i - 1][j - 1]) + 1
|
||||
}
|
||||
}
|
||||
}
|
||||
return dp[n][m]
|
||||
}
|
||||
|
||||
/* 編集距離:空間最適化した動的計画法 */
|
||||
fun editDistanceDPComp(s: String, t: String): Int {
|
||||
val n = s.length
|
||||
val m = t.length
|
||||
val dp = IntArray(m + 1)
|
||||
// 状態遷移:先頭行
|
||||
for (j in 1..m) {
|
||||
dp[j] = j
|
||||
}
|
||||
// 状態遷移:残りの行
|
||||
for (i in 1..n) {
|
||||
// 状態遷移:先頭列
|
||||
var leftup = dp[0] // dp[i-1, j-1] を一時保存する
|
||||
dp[0] = i
|
||||
// 状態遷移:残りの列
|
||||
for (j in 1..m) {
|
||||
val temp = dp[j]
|
||||
if (s[i - 1] == t[j - 1]) {
|
||||
// 2 つの文字が等しければ、その 2 文字をそのままスキップする
|
||||
dp[j] = leftup
|
||||
} else {
|
||||
// 最小編集回数 = 挿入・削除・置換の 3 操作における最小編集回数 + 1
|
||||
dp[j] = min(min(dp[j - 1], dp[j]), leftup) + 1
|
||||
}
|
||||
leftup = temp // 次の反復の dp[i-1, j-1] に更新する
|
||||
}
|
||||
}
|
||||
return dp[m]
|
||||
}
|
||||
|
||||
/* Driver Code */
|
||||
fun main() {
|
||||
val s = "bag"
|
||||
val t = "pack"
|
||||
val n = s.length
|
||||
val m = t.length
|
||||
|
||||
// 全探索
|
||||
var res = editDistanceDFS(s, t, n, m)
|
||||
println("$s を $t に変更するには最小で $res 回の編集が必要")
|
||||
|
||||
// メモ化探索
|
||||
val mem = Array(n + 1) { IntArray(m + 1) }
|
||||
for (row in mem)
|
||||
row.fill(-1)
|
||||
res = editDistanceDFSMem(s, t, mem, n, m)
|
||||
println("$s を $t に変更するには最小で $res 回の編集が必要")
|
||||
|
||||
// 動的計画法
|
||||
res = editDistanceDP(s, t)
|
||||
println("$s を $t に変更するには最小で $res 回の編集が必要")
|
||||
|
||||
// 空間最適化後の動的計画法
|
||||
res = editDistanceDPComp(s, t)
|
||||
println("$s を $t に変更するには最小で $res 回の編集が必要")
|
||||
}
|
||||
@@ -0,0 +1,125 @@
|
||||
/**
|
||||
* File: knapsack.kt
|
||||
* Created Time: 2024-01-25
|
||||
* Author: curtishd (1023632660@qq.com)
|
||||
*/
|
||||
|
||||
package chapter_dynamic_programming
|
||||
|
||||
import kotlin.math.max
|
||||
|
||||
/* 0-1 ナップサック:総当たり探索 */
|
||||
fun knapsackDFS(
|
||||
wgt: IntArray,
|
||||
_val: IntArray,
|
||||
i: Int,
|
||||
c: Int
|
||||
): Int {
|
||||
// すべての品物を選び終えたか、ナップサックに残り容量がなければ、価値 0 を返す
|
||||
if (i == 0 || c == 0) {
|
||||
return 0
|
||||
}
|
||||
// ナップサック容量を超える場合は、入れない選択しかできない
|
||||
if (wgt[i - 1] > c) {
|
||||
return knapsackDFS(wgt, _val, i - 1, c)
|
||||
}
|
||||
// 品物 i を入れない場合と入れる場合の最大価値を計算する
|
||||
val no = knapsackDFS(wgt, _val, i - 1, c)
|
||||
val yes = knapsackDFS(wgt, _val, i - 1, c - wgt[i - 1]) + _val[i - 1]
|
||||
// 2つの案のうち価値が大きいほうを返す
|
||||
return max(no, yes)
|
||||
}
|
||||
|
||||
/* 0-1 ナップサック:メモ化探索 */
|
||||
fun knapsackDFSMem(
|
||||
wgt: IntArray,
|
||||
_val: IntArray,
|
||||
mem: Array<IntArray>,
|
||||
i: Int,
|
||||
c: Int
|
||||
): Int {
|
||||
// すべての品物を選び終えたか、ナップサックに残り容量がなければ、価値 0 を返す
|
||||
if (i == 0 || c == 0) {
|
||||
return 0
|
||||
}
|
||||
// 既に記録があればそのまま返す
|
||||
if (mem[i][c] != -1) {
|
||||
return mem[i][c]
|
||||
}
|
||||
// ナップサック容量を超える場合は、入れない選択しかできない
|
||||
if (wgt[i - 1] > c) {
|
||||
return knapsackDFSMem(wgt, _val, mem, i - 1, c)
|
||||
}
|
||||
// 品物 i を入れない場合と入れる場合の最大価値を計算する
|
||||
val no = knapsackDFSMem(wgt, _val, mem, i - 1, c)
|
||||
val yes = knapsackDFSMem(wgt, _val, mem, i - 1, c - wgt[i - 1]) + _val[i - 1]
|
||||
// 2 つの案のうち価値が大きい方を記録して返す
|
||||
mem[i][c] = max(no, yes)
|
||||
return mem[i][c]
|
||||
}
|
||||
|
||||
/* 0-1 ナップサック:動的計画法 */
|
||||
fun knapsackDP(wgt: IntArray, _val: IntArray, cap: Int): Int {
|
||||
val n = wgt.size
|
||||
// dp テーブルを初期化
|
||||
val dp = Array(n + 1) { IntArray(cap + 1) }
|
||||
// 状態遷移
|
||||
for (i in 1..n) {
|
||||
for (c in 1..cap) {
|
||||
if (wgt[i - 1] > c) {
|
||||
// ナップサック容量を超えるなら品物 i は選ばない
|
||||
dp[i][c] = dp[i - 1][c]
|
||||
} else {
|
||||
// 品物 i を選ばない場合と選ぶ場合の大きい方
|
||||
dp[i][c] = max(dp[i - 1][c], dp[i - 1][c - wgt[i - 1]] + _val[i - 1])
|
||||
}
|
||||
}
|
||||
}
|
||||
return dp[n][cap]
|
||||
}
|
||||
|
||||
/* 0-1 ナップサック:空間最適化後の動的計画法 */
|
||||
fun knapsackDPComp(wgt: IntArray, _val: IntArray, cap: Int): Int {
|
||||
val n = wgt.size
|
||||
// dp テーブルを初期化
|
||||
val dp = IntArray(cap + 1)
|
||||
// 状態遷移
|
||||
for (i in 1..n) {
|
||||
// 逆順に走査する
|
||||
for (c in cap downTo 1) {
|
||||
if (wgt[i - 1] <= c) {
|
||||
// 品物 i を選ばない場合と選ぶ場合の大きい方
|
||||
dp[c] = max(dp[c], dp[c - wgt[i - 1]] + _val[i - 1])
|
||||
}
|
||||
}
|
||||
}
|
||||
return dp[cap]
|
||||
}
|
||||
|
||||
/* Driver Code */
|
||||
fun main() {
|
||||
val wgt = intArrayOf(10, 20, 30, 40, 50)
|
||||
val _val = intArrayOf(50, 120, 150, 210, 240)
|
||||
val cap = 50
|
||||
val n = wgt.size
|
||||
|
||||
// 全探索
|
||||
var res = knapsackDFS(wgt, _val, n, cap)
|
||||
println("ナップサック容量を超えない最大価値は $res")
|
||||
|
||||
// メモ化探索
|
||||
val mem = Array(n + 1) { IntArray(cap + 1) }
|
||||
for (row in mem) {
|
||||
row.fill(-1)
|
||||
}
|
||||
res = knapsackDFSMem(wgt, _val, mem, n, cap)
|
||||
println("ナップサック容量を超えない最大価値は $res")
|
||||
|
||||
// 動的計画法
|
||||
res = knapsackDP(wgt, _val, cap)
|
||||
println("ナップサック容量を超えない最大価値は $res")
|
||||
|
||||
// 空間最適化後の動的計画法
|
||||
res = knapsackDPComp(wgt, _val, cap)
|
||||
println("ナップサック容量を超えない最大価値は $res")
|
||||
}
|
||||
@@ -0,0 +1,51 @@
|
||||
/**
|
||||
* File: min_cost_climbing_stairs_dp.kt
|
||||
* Created Time: 2024-01-25
|
||||
* Author: curtishd (1023632660@qq.com)
|
||||
*/
|
||||
|
||||
package chapter_dynamic_programming
|
||||
|
||||
import kotlin.math.min
|
||||
|
||||
/* 階段登りの最小コスト:動的計画法 */
|
||||
fun minCostClimbingStairsDP(cost: IntArray): Int {
|
||||
val n = cost.size - 1
|
||||
if (n == 1 || n == 2) return cost[n]
|
||||
// 部分問題の解を保存するために dp テーブルを初期化
|
||||
val dp = IntArray(n + 1)
|
||||
// 初期状態:最小部分問題の解をあらかじめ設定
|
||||
dp[1] = cost[1]
|
||||
dp[2] = cost[2]
|
||||
// 状態遷移:小さい部分問題から大きい部分問題へ順に解く
|
||||
for (i in 3..n) {
|
||||
dp[i] = min(dp[i - 1], dp[i - 2]) + cost[i]
|
||||
}
|
||||
return dp[n]
|
||||
}
|
||||
|
||||
/* 階段昇りの最小コスト:空間最適化後の動的計画法 */
|
||||
fun minCostClimbingStairsDPComp(cost: IntArray): Int {
|
||||
val n = cost.size - 1
|
||||
if (n == 1 || n == 2) return cost[n]
|
||||
var a = cost[1]
|
||||
var b = cost[2]
|
||||
for (i in 3..n) {
|
||||
val tmp = b
|
||||
b = min(a, tmp) + cost[i]
|
||||
a = tmp
|
||||
}
|
||||
return b
|
||||
}
|
||||
|
||||
/* Driver Code */
|
||||
fun main() {
|
||||
val cost = intArrayOf(0, 1, 10, 1, 1, 1, 10, 1, 1, 10, 1)
|
||||
println("入力された階段コストのリストは ${cost.contentToString()}")
|
||||
|
||||
var res = minCostClimbingStairsDP(cost)
|
||||
println("階段を上り切る最小コストは $res")
|
||||
|
||||
res = minCostClimbingStairsDPComp(cost)
|
||||
println("階段を上り切る最小コストは $res")
|
||||
}
|
||||
@@ -0,0 +1,132 @@
|
||||
/**
|
||||
* File: min_path_sum.kt
|
||||
* Created Time: 2024-01-25
|
||||
* Author: curtishd (1023632660@qq.com)
|
||||
*/
|
||||
|
||||
package chapter_dynamic_programming
|
||||
|
||||
import kotlin.math.min
|
||||
|
||||
/* 最小経路和:全探索 */
|
||||
fun minPathSumDFS(grid: Array<IntArray>, i: Int, j: Int): Int {
|
||||
// 左上のセルなら探索を終了する
|
||||
if (i == 0 && j == 0) {
|
||||
return grid[0][0]
|
||||
}
|
||||
// 行または列のインデックスが範囲外なら、コスト +∞ を返す
|
||||
if (i < 0 || j < 0) {
|
||||
return Int.MAX_VALUE
|
||||
}
|
||||
// 左上から (i-1, j) および (i, j-1) までの最小経路コストを計算する
|
||||
val up = minPathSumDFS(grid, i - 1, j)
|
||||
val left = minPathSumDFS(grid, i, j - 1)
|
||||
// 左上隅から (i, j) までの最小経路コストを返す
|
||||
return min(left, up) + grid[i][j]
|
||||
}
|
||||
|
||||
/* 最小経路和:メモ化探索 */
|
||||
fun minPathSumDFSMem(
|
||||
grid: Array<IntArray>,
|
||||
mem: Array<IntArray>,
|
||||
i: Int,
|
||||
j: Int
|
||||
): Int {
|
||||
// 左上のセルなら探索を終了する
|
||||
if (i == 0 && j == 0) {
|
||||
return grid[0][0]
|
||||
}
|
||||
// 行または列のインデックスが範囲外なら、コスト +∞ を返す
|
||||
if (i < 0 || j < 0) {
|
||||
return Int.MAX_VALUE
|
||||
}
|
||||
// 既に記録があればそのまま返す
|
||||
if (mem[i][j] != -1) {
|
||||
return mem[i][j]
|
||||
}
|
||||
// 左と上のセルからの最小経路コスト
|
||||
val up = minPathSumDFSMem(grid, mem, i - 1, j)
|
||||
val left = minPathSumDFSMem(grid, mem, i, j - 1)
|
||||
// 左上から (i, j) までの最小経路コストを記録して返す
|
||||
mem[i][j] = min(left, up) + grid[i][j]
|
||||
return mem[i][j]
|
||||
}
|
||||
|
||||
/* 最小経路和:動的計画法 */
|
||||
fun minPathSumDP(grid: Array<IntArray>): Int {
|
||||
val n = grid.size
|
||||
val m = grid[0].size
|
||||
// dp テーブルを初期化
|
||||
val dp = Array(n) { IntArray(m) }
|
||||
dp[0][0] = grid[0][0]
|
||||
// 状態遷移:先頭行
|
||||
for (j in 1..<m) {
|
||||
dp[0][j] = dp[0][j - 1] + grid[0][j]
|
||||
}
|
||||
// 状態遷移:先頭列
|
||||
for (i in 1..<n) {
|
||||
dp[i][0] = dp[i - 1][0] + grid[i][0]
|
||||
}
|
||||
// 状態遷移: 残りの行と列
|
||||
for (i in 1..<n) {
|
||||
for (j in 1..<m) {
|
||||
dp[i][j] = min(dp[i][j - 1], dp[i - 1][j]) + grid[i][j]
|
||||
}
|
||||
}
|
||||
return dp[n - 1][m - 1]
|
||||
}
|
||||
|
||||
/* 最小経路和:空間最適化後の動的計画法 */
|
||||
fun minPathSumDPComp(grid: Array<IntArray>): Int {
|
||||
val n = grid.size
|
||||
val m = grid[0].size
|
||||
// dp テーブルを初期化
|
||||
val dp = IntArray(m)
|
||||
// 状態遷移:先頭行
|
||||
dp[0] = grid[0][0]
|
||||
for (j in 1..<m) {
|
||||
dp[j] = dp[j - 1] + grid[0][j]
|
||||
}
|
||||
// 状態遷移:残りの行
|
||||
for (i in 1..<n) {
|
||||
// 状態遷移:先頭列
|
||||
dp[0] = dp[0] + grid[i][0]
|
||||
// 状態遷移:残りの列
|
||||
for (j in 1..<m) {
|
||||
dp[j] = min(dp[j - 1], dp[j]) + grid[i][j]
|
||||
}
|
||||
}
|
||||
return dp[m - 1]
|
||||
}
|
||||
|
||||
/* Driver Code */
|
||||
fun main() {
|
||||
val grid = arrayOf(
|
||||
intArrayOf(1, 3, 1, 5),
|
||||
intArrayOf(2, 2, 4, 2),
|
||||
intArrayOf(5, 3, 2, 1),
|
||||
intArrayOf(4, 3, 5, 2)
|
||||
)
|
||||
val n = grid.size
|
||||
val m = grid[0].size
|
||||
|
||||
// 全探索
|
||||
var res = minPathSumDFS(grid, n - 1, m - 1)
|
||||
println("左上から右下までの最小経路和は $res")
|
||||
|
||||
// メモ化探索
|
||||
val mem = Array(n) { IntArray(m) }
|
||||
for (row in mem) {
|
||||
row.fill(-1)
|
||||
}
|
||||
res = minPathSumDFSMem(grid, mem, n - 1, m - 1)
|
||||
println("左上から右下までの最小経路和は $res")
|
||||
|
||||
// 動的計画法
|
||||
res = minPathSumDP(grid)
|
||||
println("左上から右下までの最小経路和は $res")
|
||||
|
||||
// 空間最適化後の動的計画法
|
||||
res = minPathSumDPComp(grid)
|
||||
println("左上から右下までの最小経路和は $res")
|
||||
}
|
||||
@@ -0,0 +1,68 @@
|
||||
/**
|
||||
* File: unbounded_knapsack.kt
|
||||
* Created Time: 2024-01-25
|
||||
* Author: curtishd (1023632660@qq.com)
|
||||
*/
|
||||
|
||||
package chapter_dynamic_programming
|
||||
|
||||
import kotlin.math.max
|
||||
|
||||
/* 完全ナップサック問題:動的計画法 */
|
||||
fun unboundedKnapsackDP(wgt: IntArray, _val: IntArray, cap: Int): Int {
|
||||
val n = wgt.size
|
||||
// dp テーブルを初期化
|
||||
val dp = Array(n + 1) { IntArray(cap + 1) }
|
||||
// 状態遷移
|
||||
for (i in 1..n) {
|
||||
for (c in 1..cap) {
|
||||
if (wgt[i - 1] > c) {
|
||||
// ナップサック容量を超えるなら品物 i は選ばない
|
||||
dp[i][c] = dp[i - 1][c]
|
||||
} else {
|
||||
// 品物 i を選ばない場合と選ぶ場合の大きい方
|
||||
dp[i][c] = max(dp[i - 1][c], dp[i][c - wgt[i - 1]] + _val[i - 1])
|
||||
}
|
||||
}
|
||||
}
|
||||
return dp[n][cap]
|
||||
}
|
||||
|
||||
/* 完全ナップサック問題:空間最適化後の動的計画法 */
|
||||
fun unboundedKnapsackDPComp(
|
||||
wgt: IntArray,
|
||||
_val: IntArray,
|
||||
cap: Int
|
||||
): Int {
|
||||
val n = wgt.size
|
||||
// dp テーブルを初期化
|
||||
val dp = IntArray(cap + 1)
|
||||
// 状態遷移
|
||||
for (i in 1..n) {
|
||||
for (c in 1..cap) {
|
||||
if (wgt[i - 1] > c) {
|
||||
// ナップサック容量を超えるなら品物 i は選ばない
|
||||
dp[c] = dp[c]
|
||||
} else {
|
||||
// 品物 i を選ばない場合と選ぶ場合の大きい方
|
||||
dp[c] = max(dp[c], dp[c - wgt[i - 1]] + _val[i - 1])
|
||||
}
|
||||
}
|
||||
}
|
||||
return dp[cap]
|
||||
}
|
||||
|
||||
/* Driver Code */
|
||||
fun main() {
|
||||
val wgt = intArrayOf(1, 2, 3)
|
||||
val _val = intArrayOf(5, 11, 15)
|
||||
val cap = 4
|
||||
|
||||
// 動的計画法
|
||||
var res = unboundedKnapsackDP(wgt, _val, cap)
|
||||
println("ナップサック容量を超えない最大価値は $res")
|
||||
|
||||
// 空間最適化後の動的計画法
|
||||
res = unboundedKnapsackDPComp(wgt, _val, cap)
|
||||
println("ナップサック容量を超えない最大価値は $res")
|
||||
}
|
||||
Reference in New Issue
Block a user