mirror of
https://github.com/krahets/hello-algo.git
synced 2026-07-12 07:26:07 +00:00
Yudong Jin
772183705e
* Add Russian docs site baseline * Add Russian localized codebase * Polish Russian code wording * Update ru code translation. * Update code translation and chapter covers. * Fix pythontutor extraction. * Add README and landing page. * placeholder of profiles * Use figures of English version * Remove chapter paperbook
58 lines
2.5 KiB
Kotlin
58 lines
2.5 KiB
Kotlin
/**
|
|
* File: subset_sum_i.kt
|
|
* Created Time: 2024-01-25
|
|
* Author: curtishd (1023632660@qq.com)
|
|
*/
|
|
|
|
package chapter_backtracking.subset_sum_i
|
|
|
|
/* Алгоритм бэктрекинга: сумма подмножеств I */
|
|
fun backtrack(
|
|
state: MutableList<Int>,
|
|
target: Int,
|
|
choices: IntArray,
|
|
start: Int,
|
|
res: MutableList<MutableList<Int>?>
|
|
) {
|
|
// Если сумма подмножества равна target, записать решение
|
|
if (target == 0) {
|
|
res.add(state.toMutableList())
|
|
return
|
|
}
|
|
// Обойти все варианты выбора
|
|
// Отсечение 2: начинать обход с start, чтобы избежать генерации повторяющихся подмножеств
|
|
for (i in start..<choices.size) {
|
|
// Отсечение 1: если сумма подмножества превышает target, немедленно завершить цикл
|
|
// Это связано с тем, что массив уже отсортирован, следующие элементы больше, и сумма подмножества точно превысит target
|
|
if (target - choices[i] < 0) {
|
|
break
|
|
}
|
|
// Попытка: сделать выбор и обновить target и start
|
|
state.add(choices[i])
|
|
// Перейти к следующему выбору
|
|
backtrack(state, target - choices[i], choices, i, res)
|
|
// Откат: отменить выбор и восстановить предыдущее состояние
|
|
state.removeAt(state.size - 1)
|
|
}
|
|
}
|
|
|
|
/* Решить задачу суммы подмножеств I */
|
|
fun subsetSumI(nums: IntArray, target: Int): MutableList<MutableList<Int>?> {
|
|
val state = mutableListOf<Int>() // Состояние (подмножество)
|
|
nums.sort() // Отсортировать nums
|
|
val start = 0 // Стартовая вершина обхода
|
|
val res = mutableListOf<MutableList<Int>?>() // Список результатов (список подмножеств)
|
|
backtrack(state, target, nums, start, res)
|
|
return res
|
|
}
|
|
|
|
/* Driver Code */
|
|
fun main() {
|
|
val nums = intArrayOf(3, 4, 5)
|
|
val target = 9
|
|
|
|
val res = subsetSumI(nums, target)
|
|
|
|
println("Входной массив nums = ${nums.contentToString()}, target = $target")
|
|
println("Все подмножества с суммой $target: res = $res")
|
|
} |