树的打印（Python）

最新推荐文章于 2024-01-18 11:58:25 发布

时越zz

最新推荐文章于 2024-01-18 11:58:25 发布

阅读量897

点赞数

本文链接：https://blog.csdn.net/qq_49644209/article/details/113790850

版权

这篇博客展示了如何利用Python的math模块计算树的深度和宽度，进而实现二叉树的两种不同方式的可视化打印。代码中定义了两个函数`print_tree`，分别通过空格对齐和居中对齐的方式展示树结构，适用于数据范围在10到20之间的列表。这两个方法有助于理解二叉树的结构和遍历方式。

摘要由CSDN通过智能技术生成

一

import math

def print_tree(lis: list, unit_width=2):
    n = len(lis) - 1

    depth = math.ceil(math.log2(n + 1))
    width = 2 ** depth - 1
    index = 1
    for i in range(depth):
        width //= 2  # 7
        pre = " " * (2 ** (4 - i) - 1) * unit_width
        tree_str = lis[index:index + 2 ** i]
        print(" " * width * unit_width, end="")
        print(pre.join(map(lambda x: "{:{}}".format(x, unit_width), tree_str)))
        index = index + 2 ** i

二

# 使用居中的方法打印
import math

'''
1.树深度公式
2.宽度=当前深度的满二叉树的所有节和叶的和
3.每个数的宽度
'''

lis = list(range(10, 20))
print(lis)


def print_tree(lis: list, unit_width=2):
    n = len(lis) - 1
    # 深度
    depth = math.ceil(math.log2(n + 1))
    width = 2 ** depth - 1
    index = 1
    for i in range(depth):
        for j in range(2 ** (i)):
            tree_str = "{:^{}}".format(lis[index], width * unit_width)
            print(tree_str, end=" " * unit_width)
            index += 1
            if index > n:
                break
        width //= 2
        print()


w = print_tree(lis)