回溯算法题（5）组合总和II

目录

组合总和II

描述

示例 1

示例 2

提示

方法：回溯

---

组合总和II

描述

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

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

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

示例 1

输入: candidates = [10,1,2,7,6,1,5], target = 8,
输出:
[
[1,1,6],
[1,2,5],
[1,7],
[2,6]
]

示例 2

输入: candidates = [2,5,2,1,2], target = 5,
输出:
[
[1,2,2],
[5]
]

提示

• 1 <= candidates.length <= 100
• 1 <= candidates[i] <= 50
• 1 <= target <= 30

方法：回溯

class Solution {
    public List<List<Integer>> res = new ArrayList<>();//存储结果
    public List<Integer> cur = new ArrayList<>();//存储当前遍历的序列
    public List<int[]> freq=new ArrayList<>();//存储数据出现的频率

    public List<List<Integer>> combinationSum2(int[] candidates, int target) {
        Arrays.sort(candidates);//对数组进行排序
        for (int num: candidates) {
            int size=freq.size();
            if (freq.isEmpty()||num!=freq.get(size-1)[0]) freq.add(new int[]{num,1});//如果当前列表没有包含该元素，直接加入
            else freq.get(size-1)[1]++;//否则出现次数加1
        }
        backTrace(target, 0);
        return res;
    }

    public void backTrace(int need, int index) {
        if (need == 0) {
            res.add(new ArrayList<>(cur));
            return;
        } else if (index == freq.size() || need < freq.get(index)[0]) return;
        else {
            backTrace(need,index+1);//不选该元素
            
            int most=Math.min(need/freq.get(index)[0],freq.get(index)[1]);//当前元素至多可以取多少个
            for (int i = 1; i <= most; i++) {
                cur.add(freq.get(index)[0]);
                backTrace(need-i*freq.get(index)[0], index+1);
            }
            for (int i = 1; i <= most; i++) cur.remove(cur.size()-1);//重置
        }
    }
}