0039组合总和 Marathon

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

candidates 中的数字可以无限制重复被选取。如果至少一个所选数字数量不同,则两种组合是唯一的。

对于给定的输入,保证和为 target 的唯一组合数少于 150 个。

示例 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]]
示例 3:

输入: candidates = [2], target = 1
输出: []
示例 4:

输入: candidates = [1], target = 1
输出: [[1]]
示例 5:

输入: candidates = [1], target = 2
输出: [[1,1]]

提示:

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combination-sum

参考:

python

# 0039.组合总和

class Solution:
    def combinationSum(self, cadidates: [int], target: int) ->[[int]]:
        res = []
        path = []

        def backTrack(cadidates, target, sum, startIndex):
            if sum > target:
                return
            if sum == target:
                return res.append(path[:])
            for i in range(startIndex, len(cadidates)):
                if sum + cadidates[i] > target: # 提前剪枝
                    return
                sum += cadidates[i]
                path.append(cadidates[i])
                backTrack(cadidates, target, sum, i) # i可以重复读取
                sum -= cadidates[i] # 回溯
                path.pop() # 回溯

        cadidates = sorted(cadidates) # 排序方便取值
        backTrack(cadidates, target, 0, 0)
        return res

golang

package backTrack

// 回溯法
func combinationSum(candidates []int, target int) [][]int {
	var path []int
	var res [][]int
	backtrackSum(0,0,target,candidates,path,&res)
	return res
}

func backtrackSum(startIndex,sum,target int, candidates,path []int, res *[][]int)  {
	// 终止条件
	if sum == target {
		tmp := make([]int, len(path))
		copy(tmp, path)
		*res = append(*res, tmp)
		return
	}
	if sum > target {return}
	// 回溯
	for i:=startIndex;i<len(candidates);i++ {
		// update path && sum
		path = append(path, candidates[i])
		sum += candidates[i]
		// recur
		backtrackSum(i, sum, target, candidates, path, res)
		// backtrack
		path = path[:len(path)-1]
		sum -= candidates[i]
	}
}

原文地址:https://www.cnblogs.com/davis12/p/15585146.html