golang二叉树 先序遍历 返回结果错误

在实现 Golang 二叉树先序遍历时，如果希望以数组形式返回所有节点值，但结果却错误，可能是由于切片保存了对基础数组的引用导致的。当将一个切片赋值给另一个时，它们都引用同一个数组。因此，函数对切片元素的更改会对调用者可见。为了解决这个问题，应该避免将切片作为参数传递给递归调用，而是让每个递归调用创建自己的结果切片。

我想以数组的形式返回所有节点的值，但是返回值错误。

type TreeNode struct {
    Left  *TreeNode
    Right *TreeNode
    Val   int
}

type BinaryTree struct {
    Root *TreeNode
}
    func PreorderRecursion(root *TreeNode, result []int) []int {
    if root == nil {
        return nil
    }
    result = append(result, root.Val)
    res1 :=PreorderRecursion(root.Left,result)
    res2 :=PreorderRecursion(root.Right,result)
    result = append(result,res1...)
    result = append(result,res2...)
    return result
}

func TestBinaryTree_PreOrder(t *testing.T) {
    tree := BinaryTree{}
    tree.Root = &TreeNode{Val: 1}
    tree.Root.Left = &TreeNode{Val: 2}
    tree.Root.Right = &TreeNode{Val: 3}

    tree.Root.Left.Left = &TreeNode{Val: 4}
    var result []int
    result =PreorderRecursion(tree.Root,result)
    fmt.Println(result,"----")
}

正确的结果应该是：1 2 4 3

但我明白了：[1 1 2 1 2 4 1 3]

解决方案

切片保存对基础数组的引用，如果您分配一个 切片到另一个，都引用同一个数组。如果一个函数需要一个 slice 参数，它对切片元素所做的更改将是 对调用者可见

查看effective go中的切片Slices

preorderrecursion 不应接受切片并更改它。这是一个方法。

func PreorderRecursion(root *TreeNode) []int {
    if root == nil {
        return nil
    }
    result := append([]int{}, root.Val)
    res1 := PreorderRecursion(root.Left)
    res2 := PreorderRecursion(root.Right)
    result = append(result, res1...)
    result = append(result, res2...)
    return result
}

问题源于您将 result 切片传递给递归调用。因此，每个递归调用都会附加上面节点的结果。您期望 1 2 4 3，但您从第一次调用中得到 1，然后从第二次调用中得到 1 2 （而不是仅 2），然后是 1 2 4 （而不是仅 4zqbendczq） b) 从第三次调用开始。

要解决此问题，您只需删除将结果切片传递给递归函数即可。该函数应该只为它所在的节点及其后代树创建一个结果切片，它不需要知道父级的结果是什么。

理论要掌握，实操不能落！以上关于《golang二叉树 先序遍历 返回结果错误》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！