二叉树的所有元素列表

Question

作为初学者，我一直在尝试在 python 中实现二叉树。 并且已经可以成功实现很多，只有一个问题是我无法返回二叉树中所有元素（ traverse() ）的列表。 我在这里使用两个类， Node和BinaryTree 。

节点 Class

class Node:
    def __init__(self, val):
        self.value = val
        self.left = None
        self.right = None

Traverse 方法返回二叉树中的所有元素。

    def traverse(self): #<-- Problem here
        tree_elements = []
        if self.left != None:
            self.left.traverse()

        tree_elements.append(self.value)

        #debug
        print(self.value, end=" ")

        if self.right != None:
            self.right.traverse()
        return tree_elements

    def addNode(self, node):
        if node.value < self.value and node.value != self.value:
            if self.left == None:
                self.left = node
            else:
                self.left.addNode(node)
        elif node.value > self.value and node.value != self.value:
            if self.right == None:
                self.right = node
            else:
                self.right.addNode(node)

    def search(self, val):
        if val == self.value:
            return "Found"

        elif val < self.value:
            if self.left != None:
                return self.left.search(val)
            else:
                return "Not Found "
        elif val > self.value:
            if self.right != None:
                return self.right.search(val)
            else:
                return "Not Found"

二叉树

class BinaryTree:
    def __init__(self):
        self.root = None

    def addVlaue(self, val):
        node = Node(val)
        if self.root == None:
            self.root = node
        else:
            self.root.addNode(node)

     def search(self, val):
         if self.root == None:
             return False
         else:
             return self.root.search(val)

    def traverse(self):
         if self.root == None:
             return "no data"
        else:
            return self.root.traverse()

问题：

traverse方法必须返回二叉树中的所有元素，但我只得到树的第一个值。

例子：

元素：100 18 46 5 65 5 31 71 94 43 #在树中输入

output 来自树：5 18 31 43 46 65 71 94 100 #预期 output

[100] #树的输出

Answer 1

tree_elements列表需要 go 沿递归调用，沿途收集每个节点。 换句话说，它必须作为参数传递给traverse调用（除了为根节点调用traverse时）。

否则，在每次 traverse 调用中都会创建并丢弃一个新列表（因为在递归期间您永远不会对traverse的返回值做任何事情），除了仅返回根元素的顶级递归调用。

试试这个实现：

def traverse(self, tree_elements=None):
    if tree_elements is None:
        tree_elements = []
   if self.left != None:
        self.left.traverse(tree_elements=tree_elements)
    tree_elements.append(self.value)
    if self.right != None:
        self.right.traverse(tree_elements=tree_elements)
    return tree_elements