Day24 Golang (回溯) 77.回溯

// 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在遍历过程中由于回溯会被覆盖