leetcode 46. 全排列 golang实现

最新推荐文章于 2024-06-21 19:27:23 发布

RRRagnaros

最新推荐文章于 2024-06-21 19:27:23 发布

阅读量511

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/weixin_42609341/article/details/104435919

leetcode 专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种基于深度遍历（回溯法）的全排列算法，通过有规律地穷举序列的所有可能组合，实现对给定数字序列的全排列生成。文章详细解释了算法的实现过程，包括如何避免重复元素、如何递归地生成所有路径以及如何存储和返回结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

给定一个没有重复数字的序列，返回其所有可能的全排列。
示例:
输入: [1,2,3]
输出:
[
[1,2,3],
[1,3,2],
[2,1,3],
[2,3,1],
[3,1,2],
[3,2,1]
]

解法：回溯法（深度遍历）

代码

func permute(nums []int) [][]int {
    l := len(nums)
    var (
        ret   = make([][]int, 0)
        paths = make([]int, 0)
        used  = make([]int, l)
    )
    if l == 0 {
        return ret
    }
    dfs(nums, paths, used, 0, l, &ret)
    return ret
}

func dfs(nums []int, paths []int, used []int, depth int, length int, ret *[][]int) {
    if depth == length {
        tmp := make([]int, len(paths))
        copy(tmp, paths)
        *ret = append(*ret, tmp)
    }
    for i := 0; i < len(nums); i++ {
        if used[i] == 1 {
            continue
        }
        paths = append(paths, nums[i])
        used[i] = 1
        dfs(nums, paths, used, depth+1, length, ret)
        used[i] = 0
        paths = paths[0 : len(paths)-1]
    }
}

思路

简单来说，整体的思路就是有规律的穷举。把所有流程平铺到一棵树上，就是一个深度遍历的过程。

从数组中选取数
从剩下的元素中继续选取作为子节点
直到深度和数组长度相同，则生成了一条路径
返回的时候将当前路径中最后一个节点拿走，从未被当作下一个节点的剩余元素中再取一个，继续以上的步骤

整个流程中用到了几个数组
ret：二维数组，用来保存已经生成的完整路径。这里要注意的是，ret在传递的时候使用了指针。保证函数内部对这个数组做改变的时候，外部也可以获取到相同的内容。容易引起问题的就是append操作，如果切片底层数组进行了扩容，那么这一次append对外部的这个切片是无效的。而我们一开始是不确定最后ret中会有多少路径的。

paths：用来保存当前进度下的所有节点。注意这里函数返回的时候要将当前的元素去掉

used：用来保存nums中每一个元素是否已经在路径中被使用。如果被使用了直接跳过。