216. 组合总和 III

mid

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

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

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

示例 1:

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

示例 2:

1
2
3
4
5
6
7
输入: 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
没有其他符合的组合了。

DFS 回溯

子集问题

k 相当于树的深度，n 是树的宽度

 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
func combinationSum3(k int, n int) [][]int {
    // k: 层数
    var res=make([][]int,0)
    var tmp=make([]int,0)
    var dfs func(layer, sum, startIndex int)
    dfs = func(layer, sum, startIndex int){
        if layer==k{
            if sum==n{
                res=append(res,append([]int{},tmp...))
            }
            return
        }
        for i:=startIndex;i<=9;i++{
            if sum+i>n{
                // 小小剪个枝
                continue
            }
            tmp=append(tmp,i)
            dfs(layer+1,sum+i,i+1)
            tmp=tmp[:len(tmp)-1]
        }
    }
    dfs(0,0,1)
    return res
}

回溯法中递归函数参数很难一次性确定下来，一般先写逻辑，需要啥参数了，填什么参数。