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实施中确定良率使用？

1 个答案: