leetcode(力扣) 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.确定回溯函数条件：

首先题目所给的条件target 肯定得要，就是目标值，另外还要一个目前进行过程中的和 sum。还要一个变量startindex来记录当前遍历的下标值，

        def backtrack(target,sum,startindex):

2.确定终止条件：
这个还是比较容易想到的，
sum大于target时，直接return。
和sum等于target时，先将当前结果加入答案集，然后再return。

			if sum > target:
                return
            if sum == target:
                res.append(temp[:])

3.循环体;

这里和上一题基本一样，只有细微变换。
老套路，回调函数前是本层到下一层操作的东西，回调函数之后是下一层返回本层要做的事。

本层到下一层，将路径上的数值加入临时列表temp，然后sum和加上该数值。
下一层回溯到本层就返回来就行了。

			for i in range(startindex,len(candidates)):
                sum += candidates[i]
                temp.append(candidates[i])
                backtrack(target,sum,i)
                sum -= candidates[i]
                temp.pop()

细节：
这道题唯一要注意的地方就是，上一道题里的startindex每次递归的时候要+1，因为上一道题每一个组合内的元素不能重复，而这道题是可以重复的，所以每次的startindex循环不需要+1了。

这里的startindex要搞清楚，他是记录 当前遍历下标从哪里开始的，在一个集合里取值 比如[2,3] 那么你取的是组合，此时 [2,3] 和[3,2]这俩是一样的，所以要记录遍历的下标，防止往回取。其实这个startindex和i的意思是一样的，让下一次的递归知道你是从哪里开始遍历的。

完整代码

class Solution:
    def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
        res = []
        temp = []
        def backtrack(target,sum,startindex):
            # 确定终止条件
            if sum > target:
                return
            if sum == target:
                res.append(temp[:])
            for i in range(startindex,len(candidates)):
                sum += candidates[i]
                temp.append(candidates[i])
                backtrack(target,sum,i)
                sum -= candidates[i]
                temp.pop()
        backtrack(target,0,0)
        return res

剪枝：

这道题的剪枝我没想到，假如给的数组为[2,5,3] target = 4。
首先对数组排序 [2,3,5] 则可以发现，第一轮取了2之后，下面的分支可以2和3和5，在取3的时候，2+3已经超过4了 所以后面的分支就没必要进行了，就在这里剪枝。

 if sum + candidates[i] > target: # 剪枝
          return 

完整代码

class Solution:
    def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
        res = []
        temp = []
        candidates.sort()
        def backtrack(target,sum,startindex):
            # 确定终止条件
            if sum > target:
                return
            if sum == target:
                res.append(temp[:])
            for i in range(startindex,len(candidates)):
                if sum + candidates[i] > target: # 剪枝
                    return 
                sum += candidates[i]
                temp.append(candidates[i])
                backtrack(target,sum,i)
                sum -= candidates[i]
                temp.pop()
        backtrack(target,0,0)
        return res