二叉树可视化程序代码
时间: 2024-10-01 09:07:13 浏览: 51
二叉树是一种常见的数据结构,其中每个节点最多有两个子节点,通常表示为“左”和“右”。为了更好地理解和调试这类数据结构,可以编写一个简单的二叉树可视化程序代码。这个程序一般会接受一个二叉树的根节点作为输入,然后递归地展示出树形结构。
以下是一个基本的Python代码示例,使用了递归的方式:
```python
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
def print_tree(node, level=0):
if node is None:
return
print(' ' * level + str(node.val))
print_tree(node.left, level+1)
print_tree(node.right, level+1)
# 使用示例
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)
print_tree(root) # 输出:1
# 2
# / \
# 4 5
# \
# 3
```
这个代码首先定义了一个`TreeNode`类来表示二叉树节点,然后`print_tree`函数通过递归遍历每个节点并增加缩进来显示层次结构。当输入空节点时,函数停止递归。
阅读全文