组合总和
题目链接:力扣题目链接
难度:中等
给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ,找出 candidates 中可以使数字和为目标数 target 的 所有 不同组合 ,并以列表形式返回。你可以按 任意顺序 返回这些组合。
candidates 中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同,则两种组合是不同的。
对于给定的输入,保证和为 target 的不同组合数少于 150 个。
示例
输入:candidates = [2,3,6,7], target = 7
输出:[[2,2,3],[7]]
解释:
2 和 3 可以形成一组候选,2 + 2 + 3 = 7 。注意 2 可以使用多次。
7 也是一个候选, 7 = 7 。
仅有这两种组合。
思路
此题和77.组合,216.组合总合Ⅲ和区别:此题没有数量要求,可以无限重复,但是总和有限制,所以间接也是有个数限制的。
对于sum已经大于target的情况,其实是依然进入了下一层递归,只是下一层递归结束判断的时候,会判断sum > target的话就返回。
其实如果已经知道下一层的sum会大于target,就没有必要进入下一层递归了。
那么可以在for循环的搜索范围上做做文章了。
对总集合排序之后,如果下一层的sum(就是本层的 sum + candidates[i])已经大于target,就可以结束本轮for循环的遍历。
回溯代码
class Solution{
List<List<Integer>> result = new ArrayList<>();
List<Integer> path = new ArrayList<>();
public List<List<Integer>> combinationSum(int[] candidates, int target) {
//先进行排序
Arrays.sort(candidates);
combinationSum(candidates,target,0,0);
return result;
}
private void combinationSum(int[] candidates,int target,int startIndex,int sum){
// 找到了数字和为 target 的组合
if(sum == target){
result.add(new ArrayList<>(path));
return;
}
for(int i=startIndex;i< candidates.length;i++){
// 如果 sum + candidates[i] > target 就终止遍历
if(sum + candidates[i] > target) break;
sum += candidates[i];
path.add(candidates[i]);
combinationSum(candidates,target,i,sum);
sum -= candidates[i];
// 回溯,移除路径 path 最后一个元素
path.remove(path.size() - 1);
}
}
}