python树的实现
时间: 2023-07-07 10:04:00 浏览: 84
python圣诞树实现
Python中可以使用类来实现树的数据结构。以下是一个示例代码:
```
class TreeNode:
def __init__(self, data):
self.data = data
self.children = []
def add_child(self, child_node):
self.children.append(child_node)
def print_tree(self):
print(self.data)
for child in self.children:
child.print_tree()
```
以上代码定义了一个树节点类`TreeNode`,其中包含节点数据`data`和子节点列表`children`。`add_child`方法用于添加子节点,`print_tree`方法用于打印整个树。
以下是一个使用示例:
```
root = TreeNode("A")
node1 = TreeNode("B")
node2 = TreeNode("C")
node3 = TreeNode("D")
node4 = TreeNode("E")
root.add_child(node1)
root.add_child(node2)
node1.add_child(node3)
node2.add_child(node4)
root.print_tree()
```
输出结果为:
```
A
B
D
C
E
```
阅读全文