以下是一个示例代码,演示了如何遍历从子节点到父节点:
class TreeNode:
def __init__(self, value):
self.value = value
self.children = []
self.parent = None
def add_child(self, child):
self.children.append(child)
child.parent = self
def traverse_to_root(node):
path = []
current_node = node
while current_node is not None:
path.append(current_node.value)
current_node = current_node.parent
# 反转路径,从子节点到父节点
path.reverse()
return path
# 创建一个树形结构
root = TreeNode("A")
node_b = TreeNode("B")
node_c = TreeNode("C")
node_d = TreeNode("D")
node_e = TreeNode("E")
node_f = TreeNode("F")
root.add_child(node_b)
root.add_child(node_c)
node_b.add_child(node_d)
node_b.add_child(node_e)
node_e.add_child(node_f)
# 遍历子节点到父节点
result = traverse_to_root(node_f)
print(result) # 输出: ['A', 'B', 'E', 'F']
在这个示例中,我们定义了一个TreeNode
类,表示树的节点。每个节点都有一个value
属性,一个children
列表保存其子节点,和一个parent
指向其父节点。
traverse_to_root
函数接受一个节点作为参数,然后从该节点开始不断迭代其父节点,将节点的值添加到path
列表中。最后,我们将path
列表反转,以得到从子节点到父节点的顺序。
在示例中,我们创建了一个树形结构,并从节点node_f
开始遍历到根节点。遍历的结果是一个包含从子节点到父节点的路径列表['A', 'B', 'E', 'F']
。
上一篇:遍历自己的链表时的内存使用情况