使用Python实现树形数据结构的操作

2023-04-11 00:00:00 python 操作数据结构

树形数据结构是由多个节点组成的，每个节点可能有一个父节点和多个子节点，可以表示层次关系。

在Python中，可以通过类来实现树形数据结构的操作。

代码示例：

class Node:
    def __init__(self, value):
        self.value = value
        self.children = []
        self.parent = None

    def add_child(self, child):
        child.parent = self
        self.children.append(child)

    def remove_child(self, child):
        child.parent = None
        self.children.remove(child)

root = Node("pidancode.com")
node1 = Node("Java")
node2 = Node("Python")
root.add_child(node1)
root.add_child(node2)
print(node1.parent.value) # 输出“pidancode.com”
root.remove_child(node2)
print(node2.parent) # 输出None

该代码中，定义了一个Node类，每个节点有value值、children子节点集合、parent父节点三个属性。其中，add_child和remove_child方法分别用于向当前节点添加子节点和从当前节点删除子节点。

创建根节点root，并向其添加两个子节点node1和node2。通过print方法可以查看node1的父节点为root，同时从root中删除了node2节点。

通过上述示例代码，可以实现树形数据结构的增删改查等操作。

相关文章