给定一个无重复元素的正整数数组 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]
}
}