// 77.组合
package main
import "fmt"
func combine(n int, k int) [][]int {
// 全局变量
res := [][]int{} //结果集
path := []int{} //每个结点的路径
// 回溯函数(递归)
var backtracking func(n, k, startindex int)
backtracking = func(n, k, startindex int) { //n为元素个数,k为结果大小,startindex为起始位置
//终止条件
if len(path) == k {
tmp := make([]int, k) //注意这里用临时变量来深拷贝path,因为path会覆盖
copy(tmp, path)
res = append(res, tmp) //将结点对应的路径加入结果集
return
}
//for循坏遍历每个结点,注意剪枝操作
for i := startindex; i <= n-(k-len(path))+1; i++ {
path = append(path, i)
backtracking(n, k, i+1) //递归,将该结点开头的所有的结果加入结果集
path = path[:len(path)-1] //回溯撤销
}
return
}
backtracking(n, k, 1)
return res
}
func main() {
res := combine(4, 2)
fmt.Printf("res: %v\n", res) //res: [[1 2] [1 3] [1 4] [2 3] [2 4] [3 4]]
}
这里在收集结果res用了深拷贝,path在遍历过程中由于回溯会被覆盖