/** * File: subset_sum_i_naive.java * Created Time: 2023-06-21 * Author: krahets (krahets@163.com) */ package chapter_backtracking; import java.util.*; public class subset_sum_i_naive { /* Алгоритм бэктрекинга: сумма подмножеств I */ static void backtrack(List state, int target, int total, int[] choices, List> res) { // Если сумма подмножества равна target, записать решение if (total == target) { res.add(new ArrayList<>(state)); return; } // Перебор всех вариантов выбора for (int i = 0; i < choices.length; i++) { // Отсечение: если сумма подмножества превышает target, пропустить этот выбор if (total + choices[i] > target) { continue; } // Попытка: сделать выбор и обновить элемент и total state.add(choices[i]); // Перейти к следующему выбору backtrack(state, target, total + choices[i], choices, res); // Откат: отменить выбор и восстановить предыдущее состояние state.remove(state.size() - 1); } } /* Решить задачу суммы подмножеств I (с повторяющимися подмножествами) */ static List> subsetSumINaive(int[] nums, int target) { List state = new ArrayList<>(); // Состояние (подмножество) int total = 0; // Сумма подмножеств List> res = new ArrayList<>(); // Список результатов (список подмножеств) backtrack(state, target, total, nums, res); return res; } public static void main(String[] args) { int[] nums = { 3, 4, 5 }; int target = 9; List> res = subsetSumINaive(nums, target); System.out.println("Входной массив nums = " + Arrays.toString(nums) + ", target = " + target); System.out.println("Все подмножества с суммой " + target + ": res = " + res); System.out.println("Обратите внимание: результат этого метода содержит повторяющиеся множества"); } }