题目:
给定一个无重复元素的数组 candidates
和一个目标数 target
,找出 candidates
中所有可以使数字和为 target
的组合。
candidates
中的数字可以无限制重复被选取。
思路:
本题是一道典型的DFS深度优先搜索+回溯+剪枝
。
- 首先设置一个List类型的集合
res
存放总结果,再设置一个cur
存放当前情况下的结果 - 因为剪枝的前提是数组有序,因此先将数组排序,然后通过DFS递归遍历数组,如果总和超过
target
,由于数组是升序遍历,后面的数肯定都不满足了,直接break
当前情况,进入下一次遍历 - 如果遍历过程中当前的目标值
target==0
,那么就将当前集合cur
加入到总的集合res
中
以下为代码+注释,结合起来会好理解一点:
// 当前遍历的集合
private List<Integer> cur = new ArrayList<>();
// 遍历的所有总结果
private List<List<Integer>> res = new ArrayList<List<Integer>>();
// 主方法
public List<List<Integer>> combinationSum(int[] candidates, int target) {
// 深度优先搜索DFS 回溯+剪枝
// 数组有序是剪枝的前提
Arrays.sort(candidates);
dfs(candidates, target, 0);
return res;
}
public void dfs(int[] arr, int target, int begin){
// 如果目标值小于等于0,进行判断,说明当前情况已经走到边界,统计进入集合
if(target == 0){
res.add(new ArrayList<>(cur));
return;
}
for(int i = begin; i < arr.length; i++){
// 剪枝 判断如果添加该值之后超出目标值,直接进行下一次循环
if(target - arr[i] < 0)
break;
cur.add(arr[i]);
dfs(arr, target - arr[i], i);
// 回溯,删除刚才添加的最后一个
cur.remove(cur.size() - 1);
}
}
笔者也在不断学习中,如有错误,欢迎指正!