leetcode 40.组合总和II Java

做题博客链接

https://blog.csdn.net/qq_43349112/article/details/108542248

题目链接

https://leetcode-cn.com/problems/combination-sum-ii/

描述

给定一个数组 candidates 和一个目标数 target ，找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的每个数字在每个组合中只能使用一次。

说明：

所有数字（包括目标数）都是正整数。
解集不能包含重复的组合。 

示例

示例 1:

输入: candidates = [10,1,2,7,6,1,5], target = 8,
所求解集为:
[
  [1, 7],
  [1, 2, 5],
  [2, 6],
  [1, 1, 6]
]

示例 2:

输入: candidates = [2,5,2,1,2], target = 5,
所求解集为:
[
  [1,2,2],
  [5]
]

初始代码模板

class Solution {
    public List<List<Integer>> combinationSum2(int[] cs, int target) {
    
    }
}

代码

推荐weiwei哥的题解：
https://leetcode-cn.com/problems/combination-sum-ii/solution/hui-su-suan-fa-jian-zhi-python-dai-ma-java-dai-m-3/

class Solution {
    List<List<Integer>> res;
    public List<List<Integer>> combinationSum2(int[] cs, int target) {
        res = new ArrayList<>();

        //重要，不仅需要排序提速，也需要排序进行去重
        Arrays.sort(cs);
        dfs(cs, target, 0, -1, false, new ArrayList<>());

        return res;
    }

    //flag表示前一个元素是否选了
    private void dfs(int[] cs, int tar, int idx, int prev, boolean flag, List<Integer> list) {
        if (tar == 0) {
            res.add(new ArrayList<>(list));
            return;
        }

        if (idx >= cs.length || tar < cs[idx]) {
            return;
        }

        //跳过当前元素
        dfs(cs, tar, idx + 1, cs[idx], false, new ArrayList<>(list));
        //如果前一个元素与当前元素相等，并且没有选取，那么当前元素也不能选取，否则会有重复结果
        if (cs[idx] == prev && !flag) {
            return;
        }

        list.add(cs[idx]);
        //选取当前元素
        dfs(cs, tar - cs[idx], idx + 1, cs[idx], true, new ArrayList<>(list));
    }
}