前言
今天这道题,沉痛的教训,思路是对的,但是漏写循环条件了。明明手写可以写出来的,但…
题目
求一个数组里所有和为 target 的组合
比如数组为 [2,2,3,4] , target = 7
则所有的组合为
[2,2,3] , [3,4]
思路
回溯剪枝,遍历所有情况,dfs模板题
sum 保存临时和,tmp保存临时的组合, index 递归到当前的深度
如果 sum == target,则当前的 tmp 是一个合法的组合,添加到答案集合里。
如果 sum + arr[index] 已经大于 target 了,则这时候不需要再加入arr[index]了,直接返回。(这步操作代表剪枝)
如果 index == arr.length了,则需要返回,因为数组会越界(这代表边界条件检查)
对每一个数组的位置都需要循环遍历,而且都需要递归深搜。搜完后还需要回溯,保证从下一个节点也可以找到所有的组合。
如果只遍历一次,那么第一个元素remove之后,就不能继续从第二个元素开始递归搜索所有的组合了。
代码
public class Solution {
//使用Set集合来去重
static Set<List<Integer>> res = new HashSet<List<Integer>>();
static List<Integer> tmp = new LinkedList<Integer>();
public static void main(String[] args) {
int[] arr = {2,2,3,2,4,1,6,5,2,7,0,1,1,1,1,1,1,1};
int target = 7;
dfs(arr,target,0,0);
printResult(res);
}
public static void dfs(int[] arr,int target,int sum,int index) {
if(sum == target){
res.add(new LinkedList<Integer>(tmp));
return;
}
if(index >= arr.length) return;
if(sum + arr[index] > target) return;
for(int i = index; i<arr.length; i++) {
tmp.add(arr[i]);
dfs(arr,target,sum+arr[i],i+1);
tmp.remove(tmp.size()-1);
}
}
public static void printResult(Set<List<Integer>> res) {
Iterator iterator = res.iterator();
while(iterator.hasNext()) {
System.out.println(iterator.next());
}
}
}
测试结果
总结
对于回溯相信大家都不陌生,但是临场突然写的时候,需要思路清晰一些再动手码代码,三思而后行。
坚持分享,坚持原创,喜欢博主的靓仔靓女们可以看看博主的首页博客!
您的点赞与收藏是我分享博客的最大赞赏!
博主博客地址: https://blog.csdn.net/weixin_43967679