题目
给你一个 无重复元素 的整数数组 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
输出: []
思路:
首先判断是组合问题, 套上经典模板
var ans [][]int
var path []int
var dfs func(int)
dfs = func(i int){
if 结束条件{
// 收割结果
return
}
// 回溯
}
dfs(0)
return ans
这里结束条件就是sum == target的时候, 但是也得考虑path不对的时候, 不能让path一直递归下去,即sum > target的时候直接return
if getSum(path) > target{
return
}
if target == getSum(path){
// 收割结果
ans = append(ans, append([]int(nil), path...))
return
}
枚举题目candidates数组中的元素,并且是可以重复的,因为元素可以重复, 所以递归直接 dfs(当前位置) 就可以 所以 递归+回溯 的操作是
for j := i; j<len(candidates); j ++ {
path = append(path, candidates[j])
dfs(j)
path = path[:len(path) - 1]
}
通过代码:
func combinationSum(candidates []int, target int) [][]int {
var ans [][]int
var path []int
var dfs func(int)
dfs = func(i int){
if getSum(path) > target{
return
}
if target == getSum(path){
// 收割结果
ans = append(ans, append([]int(nil), path...))
return
}
for j := i; j<len(candidates); j ++ {
path = append(path, candidates[j])
dfs(j)
path = path[:len(path) - 1]
}
}
dfs(0)
return ans
}
func getSum(nums []int) int{
sum := 0
for i,_ := range nums{
sum += nums[i]
}
return sum
}