LeetCode-40. 组合总和 II-Java-medium

题目链接

法一（回溯 + 剪枝）

public class Solution40 {

    private List<List<Integer>> ans;

    private List<Integer> path;

    public Solution40() {
        this.ans = new LinkedList<>();
        this.path = new LinkedList<>();
    }

    /**
     * 回溯 + 剪枝
     *
     * @param candidates
     * @param target
     * @param sum
     * @param start
     */
    void backTracking(int[] candidates, int target, int sum, int start) {
        if (sum == target) {
            ans.add(new LinkedList<>(path)); // 注意
            return;
        }
        // 如果已经知道下一层的sum会大于target，就没有必要进入下一层递归了
        for (int i = start; i < candidates.length && sum + candidates[i] <= target; i++) {
            if (i > start && candidates[i] == candidates[i - 1]) { // 跳过dfs树同一层已经使用过的元素
                continue;
            }
            sum += candidates[i];
            path.add(candidates[i]);
            backTracking(candidates, target, sum, i + 1); // 这里是i+1，每个数字在每个组合中只能使用一次
            sum -= candidates[i];
            path.remove(path.size() - 1);
        }
    }

    /**
     * 回溯 + 剪枝
     *
     * @param candidates
     * @param target
     * @return
     */
    public List<List<Integer>> combinationSum2(int[] candidates, int target) {
        Arrays.sort(candidates); // 排序，让其相同的元素都挨在一起
        backTracking(candidates, target, 0, 0);
        return ans;
    }
}

本地测试

        /**
         * 40. 组合总和 II
         */
        lay.showTitle(40);
        int[] candidates40 = new int[]{10,1,2,7,6,1,5};
        int target40 = 8;
        Solution40 sol40 = new Solution40();
        List<List<Integer>> ans40 = sol40.combinationSum2(candidates40, target40);
        arrayOpt.showIntLists(ans40);