Python 树形结构打印

class TreeNode:
    def __init__(self, value):
        self.value = value
        self.children = []


def print_tree(node, indent=0):
    print(" " * indent + str(node.value))
    for child in node.children:
        print_tree(child, indent + 1)


def ceate_tree():
    A, B, C, D, E, F = [TreeNode(x) for x in 'ABCDEF']
    A.children = [B, D]
    B.children = [C]
    D.children = [E, F]
    return A


if __name__ == '__main__':
    root = ceate_tree()
    print_tree(root)

输出

A
 B
  C
 D
  E
  F
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容