[力扣] 77.组合总和

81 阅读2分钟

题目

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

image.png