回溯,怪麻烦的。慢慢理解吧。
给定两个整数 n
和 k
,返回范围 [1, n]
中所有可能的 k
个数的组合。
你可以按 任何顺序 返回答案。
var path []int
var res [][]int
func combine(n int, k int) [][]int {
path,res = make([]int,0,k),make([][]int,0)
dfs(n,k,1)
return res
}
//n相当于树的宽度,k相当于树的深度 逐个n * 逐个k的问题
func dfs(n int, k int,start int){
//第一步判断是否到达树的叶子节点,如果是 就加入结果数组,这条边的遍历结束
if len(path) ==k{
tmp:=make([]int,k)
copy(tmp,path)
res=append(res,tmp)
return
}
//第二步,根据层数来遍历剩下的元素
for i:=start;i<=n;i++{//从左向右
if n-(k-len(path))+1<i{ break} 剪枝 组合问题 不需要重复求相同元素,需要剪纸,n-i+1是指还剩下的可组合的元素个数,k-len(path)是说当前深度与path里元素的差
//已经选择的元素个数:path.size();
//还需要的元素个数为: k - path.size();
//还需要的元素个数的起始位置 : n - (k - path.size()) + 1 < i, 选1 不考虑 2,3,4 选2 不考虑3,4 选3 不考虑 4
path = append(path,i)
dfs(n,k,i+1)//每加入一个元素,都要往上走一个元素1,2,3,4
path = path[:len(path)-1]
}
}