回溯算法——全排列

103 阅读1分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路

回溯算法框架

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//释放选择
        }
    }
}