阅读量:0
在树形结构中,TreeNode
是一个用于表示节点的类。每个节点包含一个值(如整数、字符串等)以及指向其子节点的引用。树形结构可以使用不同的存储方法,以下是两种常见的方法:
- 邻接表(Adjacency List):
邻接表是一种使用列表或字典来存储树形结构的方法。每个节点都有一个与之关联的列表,用于存储其子节点的引用。这种方法的优点是易于实现和理解,但可能会占用更多的内存。
class TreeNode: def __init__(self, value): self.value = value self.children = [] # 添加子节点 def add_child(parent, child): parent.children.append(child)
- 邻接矩阵(Adjacency Matrix):
邻接矩阵是一种使用二维数组来存储树形结构的方法。数组的每个元素表示一个节点对之间的关系。如果节点 i 是节点 j 的子节点,则矩阵中的 matrix[i][j]
值为 1(或任何非零值)。这种方法的优点是可以直接查询两个节点之间的关系,但可能会占用大量内存,特别是当树的节点数量很大时。
class TreeNode: def __init__(self, value): self.value = value class Tree: def __init__(self, num_nodes): self.num_nodes = num_nodes self.adjacency_matrix = [[0 for _ in range(num_nodes)] for _ in range(num_nodes)] # 添加子节点 def add_child(tree, parent_index, child_index): tree.adjacency_matrix[parent_index][child_index] = 1
这两种方法都可以用于表示树形结构,但具体选择哪种方法取决于问题的需求和性能要求。