【回溯剪枝】求一个数组中所有和为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

猜你喜欢

转载自blog.csdn.net/weixin_43967679/article/details/114708735