如何修复此BST实施中的产量使用？

Question

我想从树中获取值作为生成器来迭代它们。在这种情况下，yield只给了我一个值。函数print_tree按预期工作 - 它显示从最小到最大的值。如何得到类似的结果，但一个接一个地捕捉它们？这是我的BST实施：

class Node:

    def __init__(self, element):
        self.left = None
        self.right = None
        self.data = element

    def add_element(self, element):
        if self.data > element:
            if self.left is None:
                self.left = Node(element)
            else:
                self.left.add_element(element)
        else:
            if self.right is None:
                self.right = Node(element)
            else:
                self.right.add_element(element)

    def print_tree(self):
        if self.left: self.left.print_tree()
        print(self.data, end=' ')
        if self.right: self.right.print_tree()

    def get_values(self):
        if self.left: self.left.get_values()
        yield self.data
        if self.right: self.right.get_values()

    def get_structure(self):
        return (self.left.get_structure() if self.left else None, self.data, self.right.get_structure() if self.right else None)

测试代码：

from node import *

x = Node(3)
x.add_element(5)
x.add_element(1)
x.add_element(7)

print(x.get_structure())        #result: ((None, 1, None), 3, (None, 5, (None, 7, None)))
x.print_tree()                  #result: 1 3 5 7
z = x.get_values()
print(list(z))                  #result: [3]

你能解释一下我在哪里弄错了，我怎么能纠正它？

Answer 1

首先，您生成左子树中的所有元素，然后是节点值，最后是右子树中的所有元素

def get_values(self):
    if self.left:
        yield from self.left.get_values()
    yield self.data
    if self.right: 
        yield from self.right.get_values()