mirror of
https://github.com/krahets/hello-algo.git
synced 2026-07-08 13:36:06 +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
48 lines
2.5 KiB
Go
48 lines
2.5 KiB
Go
// File: subset_sum_ii.go
|
|
// Created Time: 2023-06-24
|
|
// Author: Reanon (793584285@qq.com)
|
|
|
|
package chapter_backtracking
|
|
|
|
import "sort"
|
|
|
|
/* Алгоритм бэктрекинга: сумма подмножеств II */
|
|
func backtrackSubsetSumII(start, target int, state, choices *[]int, res *[][]int) {
|
|
// Если сумма подмножества равна target, записать решение
|
|
if target == 0 {
|
|
newState := append([]int{}, *state...)
|
|
*res = append(*res, newState)
|
|
return
|
|
}
|
|
// Обойти все варианты выбора
|
|
// Отсечение 2: начинать обход с start, чтобы избежать генерации повторяющихся подмножеств
|
|
// Отсечение 3: начинать обход с start, чтобы избежать повторного выбора одного и того же элемента
|
|
for i := start; i < len(*choices); i++ {
|
|
// Отсечение 1: если сумма подмножества превышает target, немедленно завершить цикл
|
|
// Это связано с тем, что массив уже отсортирован, следующие элементы больше, и сумма подмножества точно превысит target
|
|
if target-(*choices)[i] < 0 {
|
|
break
|
|
}
|
|
// Отсечение 4: если этот элемент равен элементу слева, значит ветвь поиска повторяется, ее нужно сразу пропустить
|
|
if i > start && (*choices)[i] == (*choices)[i-1] {
|
|
continue
|
|
}
|
|
// Попытка: сделать выбор и обновить target и start
|
|
*state = append(*state, (*choices)[i])
|
|
// Перейти к следующему выбору
|
|
backtrackSubsetSumII(i+1, target-(*choices)[i], state, choices, res)
|
|
// Откат: отменить выбор и восстановить предыдущее состояние
|
|
*state = (*state)[:len(*state)-1]
|
|
}
|
|
}
|
|
|
|
/* Решить задачу суммы подмножеств II */
|
|
func subsetSumII(nums []int, target int) [][]int {
|
|
state := make([]int, 0) // Состояние (подмножество)
|
|
sort.Ints(nums) // Отсортировать nums
|
|
start := 0 // Стартовая вершина обхода
|
|
res := make([][]int, 0) // Список результатов (список подмножеств)
|
|
backtrackSubsetSumII(start, target, &state, &nums, &res)
|
|
return res
|
|
}
|