/** * File: subset_sum_i.js * Created Time: 2023-07-30 * Author: yuan0221 (yl1452491917@gmail.com) */ /* 回溯算法:子集和 I */ function backtrack(state, target, choices, start, res) { // 子集和等于 target 时,记录解 if (target === 0) { res.push([...state]); return; } // 遍历所有选择 // 剪枝二:从 start 开始遍历,避免生成重复子集 for (let i = start; i < choices.length; i++) { // 剪枝一:若子集和超过 target ,则直接结束循环 // 这是因为数组已排序,后边元素更大,子集和一定超过 target if (target - choices[i] < 0) { break; } // 尝试:做出选择,更新 target, start state.push(choices[i]); // 进行下一轮选择 backtrack(state, target - choices[i], choices, i, res); // 回退:撤销选择,恢复到之前的状态 state.pop(); } } /* 求解子集和 I */ function subsetSumI(nums, target) { const state = []; // 状态(子集) nums.sort((a, b) => a - b); // 对 nums 进行排序 const start = 0; // 遍历起始点 const res = []; // 结果列表(子集列表) backtrack(state, target, nums, start, res); return res; } /* Driver Code */ const nums = [3, 4, 5]; const target = 9; const res = subsetSumI(nums, target); console.log(`输入数组 nums = ${JSON.stringify(nums)}, target = ${target}`); console.log(`所有和等于 ${target} 的子集 res = ${JSON.stringify(res)}`);