GO-算法-回溯

一：全排列（46）

给定一个不含重复数字的数组 nums ，返回其 所有可能的全排列 。你可以 按任意顺序 返回答案。

示例 1：

输入：nums = [1,2,3]
输出：[[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]]
示例 2：

输入：nums = [0,1]
输出：[[0,1],[1,0]]
示例 3：

输入：nums = [1]
输出：[[1]]

func permute(nums []int) [][]int {
    /**
    思路分析：
    回溯方法
    **/

    //保存每一次回溯的结果，同时长度就是遍历的层数
    path := []int{}
    //记录当前数据是否出现在Path里
    visit := make(map[int]bool, len(nums))
    //返回结果
    res := [][]int{}
    //切片长度
    size := len(nums)

    res = backTrack(nums, path, visit, size, res)

    return res

}

func backTrack(nums []int, path []int, visit map[int]bool, size int, res [][]int) [][]int {
    //当遍历到最后一层，开始返回数据 此时path 已经是 123这样
    if len(path) == size {
        //存入path,这是一种情况
        // 切片底层公用数据，所以要copy
		copy(tmp, pathNums)
        tmp := make([]int, size)
        copy(tmp, path)
        res = append(res, tmp)
        return res
    }

    //递归主体，递归是深度，循环是广度
    for i:=0; i<size; i++ {
        //处理过了就跳过
        if visit[nums[i]] {
            continue
        }

        //记录处理
        visit[nums[i]] = true
        //加入数据
        path = append(path, nums[i])
        //递归
        res = backTrack(nums, path, visit, size, res)

        //回溯
        visit[nums[i]] = false
        //减去这次加的，向上返回
        path = path[:len(path)-1]
    }

    return res
}

func permute(nums []int) [][]int {
    length := len(nums)
    res, path, st := [][]int{}, []int{}, make([]bool, length)
    var dfs func(index int)
    dfs = func(index int) {
        if index == length {
            res = append(res, append([]int{}, path...))
            return
        }
        for i := 0; i < length; i++ {
            if st[i] {
                continue
            }
            path = append(path, nums[i])
            st[i] = true
            dfs(index + 1)
            st[i] = false
            path = path[:len(path)-1]
        }
    }
    dfs(0)
    return res
}