请写一段python代码要求实现一棵树的创建。
以下是一个简单的 Python 代码,用于创建一棵二叉树:
class TreeNode:
def __init__(self, data):
self.left = None
self.right = None
self.data = data
def create_tree():
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)
return root
# 测试代码
root = create_tree()
print(root.data) # 1
print(root.left.data) # 2
print(root.right.data) # 3
print(root.left.left.data) # 4
print(root.left.right.data) # 5
上述代码中,我们定义了一个 TreeNode 类,用于创建二叉树的节点。在 create_tree 函数中,我们创建了一个根节点,并为其添加左右子节点。最后,我们返回根节点,以便进行后续操作。在测试代码中,我们打印出了树中各个节点的值,以验证树的创建是否正确。
原文地址: https://www.cveoy.top/t/topic/bhq9 著作权归作者所有。请勿转载和采集!