Leetcode 40 组合总数II

题目

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

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

注意：解集不能包含重复的组合。

解题思路

  这题跟39题差不多，同样是递归+回溯，方法不再赘述。与39题不同的是，此题的数字仅能使用一次，而且有重复数字，那么就需要考虑去重，直接用 Set 保存的话，亲测会TLE。先对数组排序，在考虑下一个数的时候，需要判断当前元素是否与前一个元素相同，如果相同而且前一个元素没被用过，那么它本身也不用被考虑。因为如果相同的数前一个数没被用过，说明已经完成回溯的过程，即这个数的值已经被考虑过，再判断就是重复情况了。此题判断数组元素用没用过，用了一个 flag 数组。

代码

class Solution {
    public boolean[] flag;

    public List<List<Integer>> combinationSum2(int[] candidates, int target) {
        Arrays.sort(candidates);
        flag = new boolean[candidates.length];
        List<List<Integer>> ans = new ArrayList<>();
        dfs(candidates, target, 0, new ArrayList<>(), ans);
        return new ArrayList<>(ans);
    }

    private void dfs(int[] candidates, int target, int beginIndex, ArrayList<Integer> now, List<List<Integer>> ans) {
        if (target == 0) {
            ans.add(new ArrayList<>(now));
            return;
        }
        int length = candidates.length;
        for (int i = beginIndex; i < length; i++) {
            if (i > 0 && candidates[i - 1] == candidates[i] && !flag[i - 1]) continue;
            if (flag[i]) continue;
            if (candidates[i] > target) break;
            flag[i] = true;
            now.add(candidates[i]);
            dfs(candidates, target - candidates[i], i + 1, new ArrayList<>(now), ans);
            flag[i] = false;
            now.remove(now.size() - 1);
        }
    }
}