如何修复此 BST 实现中的 yield 使用？

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()

如何修复此 BST 实现中的 yield 使用？

How to fix yield usage in this BST implementation?

python

recurrence

yield

binary-search-tree