216. 组合总和 III
mid
找出所有相加之和为 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
没有其他符合的组合了。
DFS 回溯
子集问题
k 相当于树的深度,n 是树的宽度
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
}
回溯法中递归函数参数很难一次性确定下来,一般先写逻辑,需要啥参数了,填什么参数。