给定一个无重复元素的数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。candidates 中的数字可以无限制重复被选取。解集不能包含重复的组合。
示例 1:
输入:candidates = [2,3,6,7], target = 7,
所求解集为:[[7],[2,2,3]]
class Solution {
public List<List<Integer>> combinationSum(int[] candidates, int target) {
List<List<Integer>> res = new ArrayList<>();
Arrays.sort(candidates); // 先进行排序
backtracking(res, new ArrayList<>(), candidates, target, 0, 0);
return res;
}
public void backtracking(List<List<Integer>> res, List<Integer> path, int[] candidates, int target, int sum, int idx) {
// 找到了数字和为 target 的组合
if (sum == target) {
res.add(new ArrayList<>(path));
return;
}
for (int i = idx; i < candidates.length; i++) {
// 如果 sum + candidates[i] > target 就终止遍历
if (sum + candidates[i] > target) break;
path.add(candidates[i]);
backtracking(res, path, candidates, target, sum + candidates[i], i);
path.remove(path.size() - 1); // 回溯,移除路径 path 最后一个元素
}
}
}
一段一段拆开看
函数外边的一些变量及操作
public List<List<Integer>> combinationSum(int[] candidates, int target) {
List<List<Integer>> res = new ArrayList<>();
Arrays.sort(candidates); // 先进行排序
backtracking(res, new ArrayList<>(), candidates, target, 0, 0);
return res;
}
- 为什么需要排序?知道去重那个需要排序,但这个为什么要排序?此处的排序是为了剪枝
终止条件
回溯就是递归,递归就得有终止条件
if (sum == target) {
res.add(new ArrayList<>(path));
return;
}
- 为什么在终止条件的时候把path加到result里? 因为递归终止的时候才说明找到满足题目的集合了
- 不能光加进去不写返回
终止条件可不能写到for循环里
单层逻辑
for (int i = idx; i < candidates.length; i++) {
// 如果 sum + candidates[i] > target 就终止遍历
if (sum + candidates[i] > target) break;
path.add(candidates[i]);
backtracking(res, path, candidates, target, sum + candidates[i], i);
path.remove(path.size() - 1); // 回溯,移除路径 path 最后一个元素
}
- startIndex是下一层递归的开始位置
比如[2,3,6,7],假如第一层for中,i到元素3时进了递归,那么在本题中第二层for起始位置就是元素3,即第二层for的集合是[3,6,7] - 为什么是break而不是continue?
因为是排序后的,i之后的全是更大的元素了。用break才能达到剪枝效果 - 回溯的时候写成path.remove(path.size() - 1) ,而不是path.remove(candidates[i]),但是移的还是candidates[i]。