题目
给定一个无重复元素的数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的数字可以无限制重复被选取。
说明:
所有数字(包括 target)都是正整数。
解集不能包含重复的组合。
示例 1:
输入: candidates = [2,3,6,7], target = 7,
所求解集为:
[
[7],
[2,2,3]
]
示例 2:
输入: candidates = [2,3,5], target = 8,
所求解集为:
[
[2,2,2,2],
[2,3,3],
[3,5]
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combination-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
算法思路
这个题目是典型的回溯问题。做搜索、回溯问题画图可以很好地帮助我们理清思路,根据树形图然后再把代码写出来。
那么如何画图呢?
根据题目中的用例,画出一个树形结构图,并且在这个树形结构中要体现出递归的结构。
针对示例 1:
输入: candidates = [2, 3, 6, 7],target = 7,所求解集为: [[7], [2, 2, 3]]
递归图如下:
-
树形图以我们的目标值 target: 7作为根节点,我们可以发现其实树的每一个节点都是一个新的递归,而递归意味着对我们函数的一次回溯。而每一个节点之下都是对 candidates = [2, 3, 6, 7] 的元素逐个又一次递归,这里我们可以用 for 来实现。
-
图中被标有黑色背景的节点意味着:不用继续进行递归。我们可以发现一个规律,黑色节点中的 target 值均小于我们 candidates = [2, 3, 6, 7] 中最小的元素!这可以成为我们递归中一个判断:if(target < minNum) return;
-
图中被标有红色背景的节点意味着:target 值为 0 ,递归结束,我们已经找到了我们所求解集其中的一个组合,返回它!
-
细心的你可能也发现了,在我们返回的数组中:[2, 2, 3],[2, 3, 2],[3, 2, 2] 都是重复的。这就不满足题目:解集不能包含重复的组合。这里涉及了解决回溯问题的一个重要细节:剪枝。
❗:为了在这个问题中要实现剪枝,大家可以仔细观察下上面画出的树形结构图,假如我们可以保证:在每一次的递归之中,节点所选的元素都大于等于父节点所选元素,那么剪枝就完成了!要实现这个剪枝的过程,在递归之前我们可以先将 candidates 数组进行排序(为了简便,在给出的例子中数组是有序的,实际问题中可不一定)。
到此我们的完整算法就已经完成了,处理后的树形结构图如下:
代码实现
/**
* @param {number[]} candidates
* @param {number} target
* @return {number[][]}
*/
var combinationSum = function(candidates, target) {
candidates.sort((a,b) => a-b); // 先对数组进行排序,方便剪枝
let combination = [];
let path = [];
const len = candidates.length, minNum = candidates[0];
getCombination(candidates, target, 0, path); // 递归函数
function getCombination(candidates, target, start, path){
if(target == 0) return combination.push(path.slice()); // 找到组合,返回到解集之中
if(target < minNum) return;
for(let i = start; i < len; i++){
path.push(candidates[i]);
// 对 i节点进行递归,注意 target要减少噢
getCombination(candidates, target-candidates[i], i, path);
path.pop();
}
}
return combination;
};