【回溯剪枝】求一个数组中所有和为target的组合

---

前言

今天这道题，沉痛的教训，思路是对的，但是漏写循环条件了。明明手写可以写出来的，但…

---

题目

求一个数组里所有和为 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