而不是调用self.left / right.in_order_list(),而是调用self.left / right.pre_order_list().
为了完成你想要做的事情,生成器函数可能比在列表中累积值更好(消耗更少的内存和更多的pythonic):
class Tree(object):
...
def in_order(self):
if self.left is not None:
for value in self.left.in_order():
yield value
yield self.value #
if self.right is not None:
for value in self.right.in_order():
yield value
...
tree = Tree(...)
in_order_values = list(tree.in_order())
这样,如果您只想迭代值,则不必创建列表:
for value in tree.in_order():
...
该算法的工作原理如下:生成器首先沿着每个节点的左分支递归下降,直到它遇到没有左子节点的那个.然后它产生当前节点的值.之后,它在右侧子节点上执行相同操作,但从当前节点开始,而不是从根节点开始.如果我们假设树中没有循环而没有无限分支,那么肯定会有叶节点,即没有左或右子节点的节点. IOW节点,达到两个基本情况(self.left / right为None).因此,递归调用将返回,希望在我们内存不足或达到堆栈限制之前.
self.left / right.in_order()的循环是必要的,因为对in_order()的调用返回的是生成器,因此是名称生成器函数.返回的发电机必须以某种方式耗尽,例如,通过一个循环.在循环体中,我们将值重新生成一个级别,在那里它们再次重新生成,直到它们达到顶级.我们使用这些值.
如果你想自己检索节点而不是只检索它们的值字段,你可以这样做:
class Tree(object):
...
def in_order(self):
if self.left is not None:
for node in self.left.in_order():
yield node
yield self #
if self.right is not None:
for node in self.right.in_order():
yield node
您可能希望这样做,因为您不仅可以访问节点的值:
for node in tree.in_order():
do_something_with(node.value)
你也可以用节点做任何你想做的事情:
for node in tree.in_order():
whatever(node.some_other_attribute)