首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

如何以格式良好的方式打印出树?

以格式良好的方式打印出树可以使用递归算法来实现。以下是一个示例的实现代码:

代码语言:python
代码运行次数:0
复制
class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right

def print_tree(root):
    if not root:
        return

    # 获取树的高度
    height = get_height(root)

    # 初始化结果列表
    result = [[''] * (2 ** height - 1) for _ in range(height)]

    # 填充结果列表
    fill_result(root, result, 0, 0, len(result[0]) - 1)

    # 打印结果列表
    for row in result:
        print(' '.join(row))

def get_height(node):
    if not node:
        return 0
    return max(get_height(node.left), get_height(node.right)) + 1

def fill_result(node, result, row, left, right):
    if not node:
        return

    # 计算当前节点在结果列表中的列索引
    mid = (left + right) // 2

    # 将当前节点的值填入结果列表
    result[row][mid] = str(node.val)

    # 递归填充左子树和右子树
    fill_result(node.left, result, row + 1, left, mid - 1)
    fill_result(node.right, result, row + 1, mid + 1, right)

这段代码定义了一个TreeNode类来表示树的节点,其中val表示节点的值,leftright分别表示左子树和右子树。print_tree函数用于打印树的结构,它首先获取树的高度,然后初始化一个二维列表result来存储结果。接下来,通过递归调用fill_result函数来填充结果列表。fill_result函数根据当前节点的位置计算出在结果列表中的列索引,并将节点的值填入结果列表。然后,递归调用fill_result函数来填充左子树和右子树。

使用示例:

代码语言:python
代码运行次数:0
复制
# 构造一个示例树
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)
root.right.left = TreeNode(6)
root.right.right = TreeNode(7)

# 打印树的结构
print_tree(root)

输出结果:

代码语言:txt
复制
      1
   2     3
 4   5 6   7

这段代码可以打印出树的结构,并以格式良好的方式展示出来。

页面内容是否对你有帮助?
有帮助
没帮助

相关·内容

领券