216. 组合总和 III

题目描述

找出所有相加之和为 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

解法一：回溯

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26


func combinationSum3(k int, n int) [][]int {
    arr := []int{1, 2, 3, 4, 5, 6, 7, 8, 9}
    var tmp []int
    // cnt is the number of selected number
    var ans [][]int
    var backtracking func(idx, cnt, sum int)
    backtracking = func(idx, cnt, sum int) {
        if cnt == k {
            if sum == n {
                replica := make([]int, k)
                copy(replica, tmp)
                ans = append(ans, replica)
            }
            return
        }
        if sum > n || idx >= 9 {
            return
        }
        tmp = append(tmp, arr[idx])
        backtracking(idx+1, cnt+1, sum+arr[idx])
        tmp = tmp[:len(tmp)-1]
        backtracking(idx+1, cnt, sum)
    }
    backtracking(0, 0, 0)
    return ans
}