以格式良好的方式打印出树可以使用递归算法来实现。以下是一个示例的实现代码:
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
表示节点的值,left
和right
分别表示左子树和右子树。print_tree
函数用于打印树的结构,它首先获取树的高度,然后初始化一个二维列表result
来存储结果。接下来,通过递归调用fill_result
函数来填充结果列表。fill_result
函数根据当前节点的位置计算出在结果列表中的列索引,并将节点的值填入结果列表。然后,递归调用fill_result
函数来填充左子树和右子树。
使用示例:
# 构造一个示例树
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)
输出结果:
1
2 3
4 5 6 7
这段代码可以打印出树的结构,并以格式良好的方式展示出来。
领取专属 10元无门槛券
手把手带您无忧上云