【LeetCode】40. 组合总和 II

117 阅读1分钟

「这是我参与2022首次更文挑战的第29天,活动详情查看:2022首次更文挑战」。

题目

给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ,找出 candidates 中可以使数字和为目标数 target 的 所有 不同组合 ,并以列表形式返回。你可以按 任意顺序 返回这些组合。

candidates 中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同,则两种组合是不同的。

对于给定的输入,保证和为 target 的不同组合数少于 150 个。

示例 1

输入:candidates = [2,3,6,7], target = 7
输出:[[2,2,3],[7]]
解释:
2 和 3 可以形成一组候选,2 + 2 + 3 = 7 。注意 2 可以使用多次。
7 也是一个候选, 7 = 7 。
仅有这两种组合。

示例 2

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

示例 3

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

提示

  • 1 <= candidates.length <= 30
  • 1 <= candidates[i] <= 200
  • candidate 中的每个元素都 互不相同
  • 1 <= target <= 500

题解

思路

回溯法用于寻找所有的可行解,通常用一个树来表示,如下图所示。每一个叶子节点都代表着一个组合,当然组合是否是有效的需要根据题意来分析。

在本题中,组合中的元素之和需要等于target,因此并不是所有的叶子节点都是合法的组合。

  • 从根节点出发,每到达一个节点后,会根据当前状态和题意作出一系列选择来更新路径(path)中的元素。
    • 状态路径(path)在本题中表示某个组合中的元素。
    • 在本题中我们会从当前候选的数字集合中选中一个数字放入path中。
  • 当到达叶节点时,表示寻找到了一个组合。
    • 这时路径之和如果等于target时,说明找到了一个符合题意的组合,即绿色的叶子节点。
  • 到达叶节后还会进行回溯,即返回到上一个节点,然后进行新的选择来计算另外一种组合。
    • path就像是一个栈结构,返回到上一个节点的同时,还需要删除path中最后添加进去的元素。
  • 当遍历完整个树的叶子节点后,就表示寻找到了所有的可能的组合了。

需要特别注意一下,因为题中有一个要求:同一个数字可以无限制重复被选取,所以我们可以重复地使用候选数字集合中第一个数字,即图中2->2->2->2这样的路径。

这样我们就必须对树进行剪枝操作,不然就会出现栈溢出的情况。如果在路径中走到下一个节点时路径和大于target,那么就没必要对其进行搜索了,即对当前路径进行剪枝操作。

代码

class Solution {
    List<List<Integer>> result = new ArrayList<>();
    public List<List<Integer>> combinationSum(int[] candidates, int target) {
        Arrays.sort(candidates);
        backtracking(candidates, target, 0, new ArrayList<>(), 0);
        return result;
    }
    public void backtracking(int[] candidates, int target, int start, List<Integer> path, int sum) {
        if (sum == target) { 
            result.add(new ArrayList<>(path));
            return;
        }
        for (int i = start; i < candidates.length && (sum + candidates[i] <= target); i++) {
            path.add(candidates[i]);
            backtracking(candidates, target, i, path, sum + candidates[i]);
            path.remove(path.size() - 1);
        }
    }
}

结语

业精于勤,荒于嬉;行成于思,毁于随。