LeetCode 热题 HOT 100之组合总和

题目39：给你一个 无重复元素 的整数数组 candidates 和一个目标整数 target ，找出 candidates 中可以使数字和为目标数 target 的 所有不同组合 ，并以列表形式返回。你可以按 任意顺序 返回这些组合。
candidates 中的 同一个 数字可以 无限制重复被选取 。如果至少一个数字的被选数量不同，则两种组合是不同的。
对于给定的输入，保证和为 target 的不同组合数少于 150 个。
方案：使用简单回溯法会发生重复，因此需要限制搜索的起点，

class Solution:
    def dfs(self, candidates, ans, target, tmp, index):
        if target < 0:
            return
        if target == 0:
            ans.append(tmp)
            return
        for idx in range(index, len(candidates)):
            self.dfs(candidates, ans, target - candidates[idx],
                     tmp + [candidates[idx]], idx)

    def combinationSum(self, candidates: List[int],
                       target: int) -> List[List[int]]:
        ans = []
        candidates.sort()
        self.dfs(candidates, ans, target, [], 0)
        return ans

当然优化代码，还需要对剪枝进行优化，在搜索路径的过程中如果目标值小于零，则可以停止这条路径。

class Solution:
    def dfs(self, candidates, ans, target, tmp, index):

        if target == 0:
            ans.append(tmp)
            return
        for idx in range(index, len(candidates)):
            if target - candidates[idx] < 0:
                break
            self.dfs(candidates, ans, target - candidates[idx],
                     tmp + [candidates[idx]], idx)

    def combinationSum(self, candidates: List[int],
                       target: int) -> List[List[int]]:
        ans = []
        candidates.sort()
        self.dfs(candidates, ans, target, [], 0)
        return ans

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/combination-sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。