LeetCode(python)39.组合总和

题目

给你一个 无重复元素 的整数数组 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
2 <= candidates[i] <= 40
candidates 的所有元素 互不相同
1 <= target <= 40

思路

对于数组中的每个数字，都有选or不选两种可能；同时可以选任意个。很明显的递归思路啊~

1.递归终止条件：当选取的数字总和≥target，停止；此时如果恰好等于，则存进答案

2.单层递归逻辑：遍历数组中的数字，加入预选集中，递归；

注意：同一个数字可以无限制重复选取，但是注意遍历的下标，不能超过当前处理的数字的下标

代码

python 复制代码

class Solution:
    def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
        ans = []
        temp = []

        def backtrack(s,index):
            if s > target:
                return
            if s == target:
                ans.append(temp[:])  # 注意不加[:]存的是引用，则temp的值会随着递归变化，全为空
                return
            for i in range(index,len(candidates)):   # 从index（当前处理的数字开始遍历）
                s += candidates[i]   
                temp.append(candidates[i])
                backtrack(s,i)
                s -= candidates[i]
                temp.pop()
        backtrack(0,0)
        
        return ans