如何使用生成器迭代树的叶子

Question

问题：

我有一个 trie 并且我想返回存储的信息在其中。有些叶子有信息（设置为值> 0），有些叶子没有。我只想返回那些有价值的叶子。

因为在所有 trie 中，每个节点上的叶子数量都是可变的，并且每个值的键实际上由到达每个叶子所需的路径组成。

我正在尝试使用生成器来遍历树后序，但我无法让它工作。我做错了什么？

我的模块：

class Node():
    '''Each leaf in the trie is a Node() class'''
    def __init__(self):
        self.children = {}
        self.value = 0

class Trie():
    '''The Trie() holds all nodes and can return a list of their values'''
    def __init__(self):
        self.root = Node()
    def add(self, key, value):
        '''Store a "value" in a position "key"'''
        node = self.root
        for digit in key:
            number = digit
            if number not in node.children:
                node.children[number] = Node()
            node = node.children[number]
        node.value = value
    def __iter__(self):
        return self.postorder(self.root)
    def postorder(self, node):
        if node:
            for child in node.children.values():
                self.postorder(child)
            # Do my printing / job related stuff here
            if node.value > 0:
                yield node.value

示例使用：

>>trie = Trie()
>>trie.add('foo', 3)
>>trie.add('foobar', 5)
>>trie.add('fobaz', 23)

>>for key in trie:
>>....print key
>>
3
5
23

我知道给出的示例很简单，可以使用任何其他数据结构来解决。然而，对于该程序来说，使用 trie 非常重要，因为它对于数据访问模式非常有益。

感谢您的帮助！

注意：我在代码块中省略了换行符，以便能够更轻松地复制粘贴。

Answer 1

更改

self.postorder(child)

为

for n in self.postorder(child):
    yield n

似乎可以使其工作。

PS 省略空白行以方便剪切和粘贴对您非常有帮助。粘贴 ：）