组合总数II[每个元素只能用一次 + 去重复解集]

前言

求排列组合，一半都是回溯搜索，利用技巧来降低递归栈深度，利用剪枝来降低运行时间。组合总数II相比直接求组合和为target，不同之处在于每个元素只能用一次，且需要去重（由于数组中有相同元素多次选择导致）。

一、组合总数II

二、回溯搜索+去重

package everyday.medium;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

// 组合总数II
public class CombinationSum3 {
    /*
    target:从candidates数组里面选数字，让其和为target，每个数字在组合中只能使用一次。
    同样是标准的回溯，但是不同点在于下一次就是cur加1，而不是cur，因为每个数字只能被选一次。
     */
    public List<List<Integer>> combinationSum2(int[] candidates, int target) {
        List<List<Integer>> ans = new ArrayList<>();
        List<Integer> el = new ArrayList<>();
        // 标准回溯通用技巧1：把可排序的数组排序，让大数先行，然后配合剪枝，减少递归栈深度。
        Arrays.sort(candidates);
        // 回溯搜索去填充符合条件的组合。
        dfs(candidates, candidates.length - 1, 0, target, el, ans);
        // 返回填充好的组合。
        return ans;
    }

    private void dfs(int[] candidates, int cur, int sum, int target, List<Integer> el, List<List<Integer>> ans) {
        // 当sum 大于等于 target 时，结束递归，递归出口1。
        if (sum >= target) {
            if (sum == target) ans.add(new ArrayList<>(el));
            return;
        }
        // 递归出口2：元素用完了。
        // bug2:有相同元素时，会导致多个重复组合。
        int pre = 1 << 31;
        for (int i = cur; i >= 0; i--) {
            if (pre == candidates[i]) continue;
            // 标准回溯。
            el.add(candidates[i]);
            // bug1:cur位经常写出cur - 1，应该写i - 1，选择每一轮的变量i.
            dfs(candidates, i - 1, sum + candidates[i], target, el, ans);
            el.remove(el.size() - 1);
            // 更新pre，防止相同元素构成重复集合。
            pre = candidates[i];
        }
    }
}

总结

1）标准回溯搜索练习。
2）回溯的技巧与剪枝练习。
3）如何去重，以及一些回溯时常见小bug。

参考文献

[1] LeetCode 组合总数II