使用 __iter__ 读取二叉树

Question

我有一个标准的二叉树，读起来像
1
/ \
2 3
/ \ / \
4 5 6 7

我需要通过导入我的树类来读取二叉树，创建这棵树，然后使用 for 循环读取它，如下所示：[4, 2, 5, 1, 6, 3, 7]

我已经制作了树，我的程序将生成一个具有任意数量数字的类似树。我的问题出在方法上。

def iter（自我）：

到目前为止我有：

def __iter__(self):
    if self.left:
        self.left.__iter__()
    yield self
    if self.right:
        self.right.__iter__()

但是当我在树对象上运行 for 循环时：

对于树中的项目： print("{}: {}").format(item.name, item.height())

它只以正确的高度打印我尝试中的第一个节点。

我的问题基本上是，如何使用递归打印这个二叉树？

Answer 1

在 Python >= 3.3 中，可以将 yield from x 语法用于递归迭代器：

def __iter__(self):
    if self.left:
        yield from self.left
    yield self
    if self.right:
        yield from self.right

编辑：确认支持 Python 3.3。

Answer 2

集合中的下一个元素应该由__next__()返回：What is the interface for python iterators?

__iter__() 应该返回一个迭代器，一个定义__next__() 方法的对象。 yield 用于生成器，这与迭代器的概念完全不同，但您可以使用生成器函数（因为它是一个迭代器），即：

class A:
    def __init__(self, val, a=None, b=None):
        self.left = a
        self.right = b
        self.val = val

    def __iter__(self):
        def generator():
            if self.left:
                for leaf in self.left:
                    yield leaf
            yield self
            if self.right:
                for leaf in self.right:
                    yield leaf
        return generator()

for i in A(3, A(5, A(2), A(1)), A(4)):
    print(i.val)

Answer 3

没有尝试过，但可能是这样的：

def __iter__(self):
    if self.left:
        for k in self.left:
            yield k
    yield self
    if self.right:
        for k in self.right:
            yield k

但先检查一下。