larscheng / algo

0 stars 0 forks source link

【Day 81 】2024-02-04 - 40 组合总数 II #81

Open larscheng opened 9 months ago

larscheng commented 9 months ago

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

larscheng commented 9 months ago

思路

与39题的区别在于 candidates数组中的元素不可重复使用 所以在进行递归时,下一个待选择元素需要+1

观察示例1可以发现,因为candidates数组存在重复元素,所以结果集中会出现相同组合,所以还需要进行去重剪枝处理 画出树形图,可以观察到同一层中的元素不可以重复,同一个元素仅一次有效

代码

class Solution {

    public List<List<Integer>> combinationSum2(int[] candidates, int target) {
        int length = candidates.length;
        List<List<Integer>> res = new ArrayList<>();

        Arrays.sort(candidates);
        Deque<Integer> queue = new ArrayDeque<>();
        dfs(candidates,0,target,length,queue,res);
        return res;
    }

    private void dfs(int[] candidates, int begin, int target, int length, Deque<Integer> queue, List<List<Integer>> res) {
        if (target==0){
            res.add(new ArrayList<>(queue));
            System.out.println();
            return;
        }

        for (int i = begin; i < length; i++) {
            if (target - candidates[i] < 0) {
                break;
            }
            //同一层相同元素进行剪枝
            if (i > begin && candidates[i] == candidates[i - 1]) {
                continue;
            }
            queue.addLast(candidates[i]);
            //递归元素+1
            dfs(candidates, i + 1, target - candidates[i], length, queue, res);

            queue.removeLast();
        }
    }
}