DFS与回溯专题：组合总和II

    DFS与回溯专题：组合总和II

题目链接: 组合总和II

参考题解：代码随想录

题目描述

代码纯享版

class Solution {

    public List<List<Integer>> list_all = new ArrayList();
    public List<Integer> list = new ArrayList();

    public List<List<Integer>> combinationSum2(int[] candidates, int target) {
        Arrays.sort(candidates);
        backtrack(candidates, target, 0, 0);
        return list_all;
    }

    void backtrack(int[] candidates, int target, int start, int sum){

        if(sum == target){
			list_all.add(new ArrayList(list));
            return;
        }

        for(int i = start; i < candidates.length; i++){

            //不剪枝通过不了
            if(sum + candidates[i] > target) return;
            if (i > start && candidates[i] == candidates[i - 1]) {
                continue;
            }

            list.add(candidates[i]);
            backtrack(candidates, target, i + 1, sum + candidates[i]);
            list.remove(list.size() - 1);
        }
    }
}

代码逐行解析版

class Solution {

    public List<List<Integer>> list_all = new ArrayList();
    public List<Integer> list = new ArrayList();

    public List<List<Integer>> combinationSum2(int[] candidates, int target) {
        Arrays.sort(candidates);  //对数组进行从小到大排序，方便后面的遍历
        backtrack(candidates, target, 0, 0); //回溯
        return list_all; //返回结果
    }

    void backtrack(int[] candidates, int target, int start, int sum){

        if(sum == target){ //当数字和等于目标数时
            list_all.add(new ArrayList(list)); //将list列表添加到list_all列表
            return;
        }

        for(int i = start; i < candidates.length; i++){

            //不进行这两个if的剪枝通过不了
            if(sum + candidates[i] > target) return; //出现大于目标数时，直接退出
            //同层！！！节点如果有多个相同，只需使用第一个即可，其他直接跳过，防止大量重复计算
            if (i > start && candidates[i] == candidates[i - 1]) { 
                continue;
            }

            list.add(candidates[i]); //将数组元素添加到list列表中
            //进入下一层的递归，start=i+1，sum=sum + candidates[i]
            backtrack(candidates, target, i + 1, sum + candidates[i]); 
            list.remove(list.size() - 1); //回溯，删掉list列表中最后一个元素
        }
    }
}