leetcode-39. 组合总和

题目

给定一个无重复元素的数组 candidates 和一个目标数 target ，找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的数字可以无限制重复被选取。

说明：

所有数字（包括 target）都是正整数。
解集不能包含重复的组合。 

示例 1：

输入：candidates = [2,3,6,7], target = 7,
所求解集为：
[
  [7],
  [2,2,3]
]

示例 2：

输入：candidates = [2,3,5], target = 8,
所求解集为：
[
  [2,2,2,2],
  [2,3,3],
  [3,5]
]

提示：

1 <= candidates.length <= 30
1 <= candidates[i] <= 200
candidate 中的每个元素都是独一无二的。
1 <= target <= 500

解题思路

回溯，为了避免重复，每次考虑从当前数字开始（包含当前数字）向后的candidates

代码

class Solution:
    def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
        output = []
        def backtrace(backtrace_candidates: list, target: int, ans: list) -> None:
            if target == 0:
                output.append(ans[:])
                return
            if target < 0:
                return
            for index, each_candidate in enumerate(backtrace_candidates):
                backtrace(backtrace_candidates[index:], target - each_candidate, ans + [each_candidate])
            return
        backtrace(candidates, target, [])
        return output