Skip to the content.

leetcode [40] 组合总和 II


Contact me:

Blog -> https://cugtyt.github.io/blog/index
Email -> cugtyt@qq.com
GitHub -> Cugtyt@GitHub


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

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

说明:

所有数字(包括目标数)都是正整数。 解集不能包含重复的组合。

示例 1:

输入: candidates = [10,1,2,7,6,1,5], target = 8,
所求解集为:
[
  [1, 7],
  [1, 2, 5],
  [2, 6],
  [1, 1, 6]
]

示例 2:

输入: candidates = [2,5,2,1,2], target = 5,
所求解集为:
[
  [1,2,2],
  [5]
]

dfs回溯,终止条件是当前值等于目标值,或者加上候选值后大于目标值。如果没有终止,将当前值加入序列,继续dfs。注意,由于不让重复,所以加入一个位置坐标表示后续的起始位置,并且需要判断当前候选值和前一个候选值是否一样,如果一样则跳过

class Solution:
    def combinationSum2(self, candidates: List[int], target: int) -> List[List[int]]:
        candidates = sorted(candidates)
        ans = []
        def dfs(seq, cur, pos):
            nonlocal target, candidates, ans
            if cur == target:
                ans.append(seq)
            
            for i in range(pos, len(candidates)):
                if candidates[i] > target or cur + candidates[i] > target:
                    break
                if i > pos and candidates[i] == candidates[i - 1]:
                    continue
                dfs(seq + [candidates[i]], cur + candidates[i], i + 1)
        dfs([], 0, 0)
        return ans