我想以数组的形式返回所有节点的值,但是返回值是错误的。
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]
答案 0 :(得分:0)
问题源于您将result
切片传递给递归调用。因此,每个递归调用都将追加来自上面节点的结果。您期望1 2 4 3
,但从第一个电话获得1
,然后从第二个电话获得1 2
(而不是2
),然后从1 2 4
(而不是仅来自第三次通话的4
)。
要解决此问题,您只需删除将结果切片传递给递归函数即可。该功能只应为其所在的节点及其后代树创建一个结果切片,而不必知道父节点的结果是什么。
答案 1 :(得分:0)
切片包含对基础数组的引用,如果您分配一个, 切片到另一个,都引用相同的数组。如果一个函数需要一个 slice参数,它对slice元素所做的更改将是 呼叫者可见
查看有效执行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
}