216. 组合总和 III

leecode原题

题目

找出所有相加之和为 n 的 k 个数的组合，且满足下列条件：

• 只使用数字1到9
• 每个数字 最多使用一次

返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次，组合可以以任何顺序返回。

示例

示例 1:

输入: k = 3, n = 7
输出: [[1,2,4]]
解释:
1 + 2 + 4 = 7
没有其他符合的组合了。

示例 2:

输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]
解释:
1 + 2 + 6 = 9
1 + 3 + 5 = 9
2 + 3 + 4 = 9
没有其他符合的组合了。

示例 3:

输入: k = 4, n = 1
输出: []
解释: 不存在有效的组合。
在[1,9]范围内使用4个不同的数字，我们可以得到的最小和是1+2+3+4 = 10，因为10 > 1，没有有效的组合。

提示：

• 2 <= k <= 9
• 1 <= n <= 60

解题思路

思路

这道题跟77.组合很像，主要只是多了目标和的限制.

实现

源码

var (
    path = make([]int, 0)   // 符合条件的结果
    res  = make([][]int, 0) // 存放最终的结果集
)

func backtracking(k int, targetNum int, startIndex int, sum int) {
    if sum > targetNum { // 剪枝操作
        return
    }
    if len(path) == k {
        if sum == targetNum {
            tmp := make([]int, k)
            copy(tmp, path)
            res = append(res, tmp)
        }
        return
    }
    for i := startIndex; i <= 9-(k-len(path))+1; i++ { //裁枝
        path = append(path, i) // 处理
        sum += i               // 处理
        backtracking(k, targetNum, i+1, sum)
        sum -= i                  //回溯
        path = path[:len(path)-1] //回溯
    }
}

func combinationSum3(k int, n int) [][]int {
    path = make([]int, 0)  // 符合条件的结果
    res = make([][]int, 0) // 存放最终的结果集
    backtracking(k, n, 1, 0)
    return res
}