本文已参与「新人创作礼」活动,一起开启掘金创作之路
回溯算法框架
if 退出条件 {
//将得到的值,保存到变量中
}
for _ , 变量 := range 所有变量 {
if 变量符合条件 {
将变量设为不符合条件,防止回溯之后重复选择
backtrace() //选择
将变量的不符合的释放,可以再次选择
}
}
例题
给定一个不含重复数字的数组 nums ,返回其 所有可能的全排列 。你可以 按任意顺序 返回答案。
示例 1:
输入:nums = [1,2,3] 输出:[[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]]
来源:力扣(LeetCode) 链接:leetcode-cn.com/problems/pe… 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
将给定的数组进行全排列。使用回溯算法,我们可以讲所有的排列得到,然后存储到变量中。
var des [][]int
func permute(nums []int) [][]int {
des = make([][]int , 0)//初始化
backtrace(nums , make([]int , 0) , make([]bool , len(nums)))//回溯
return des//返回
}
func backtrace(nums , path []int , chose []bool ) {
if len(path) == len(nums) {//如果排列之后的切片和给定的数组长度相同,则返回,说明已经排列所有的元素
temp := make([]int , len(nums))
copy(temp , path)
des = append(des , temp)//将排列好的切片存储
}
for i := 0 ; i < len(chose) ; i ++ {
if !chose[i] { //遍历切片的元素,如果没有被选择,则选择它
chose[i] = true //标记选择,防止重复选择
backtrace(nums , append(path , nums[i]) , chose)
chose[i] = false//释放选择
}
}
}