实现深度优先方式遍历树

以深度优先方式遍历树形节点的生成器。

class Node:    def __init__(self, value):        self._value = value        self._children = []    def __repr__(self):        return 'Node({!r})'.format(self._value)    def add_child(self, node):        self._children.append(node)    def __iter__(self):        return iter(self._children)    def depth_first(self):        yield self        for c in self:            yield from c.depth_first()# Exampleif __name__ == '__main__':    root = Node(0)    child1 = Node(1)    child2 = Node(2)    root.add_child(child1)    root.add_child(child2)    child1.add_child(Node(3))    child1.add_child(Node(4))    child2.add_child(Node(5))    for ch in root.depth_first():        print(ch)

# Outputs Node(0), Node(1), Node(3), Node(4), Node(2), Node(5)

在这段代码中，depth_first() 方法简单直观。 它首先返回自己本身并迭代每一个子节点并 通过调用子节点的 depth_first() 方法(使用 yield from 语句)返回对应元素。

来自：Python cookbook