📅  最后修改于: 2023-12-03 15:39:41.095000             🧑  作者: Mango
在树结构中,叶节点是指没有子节点的节点。在某些场景下,我们需要定位一个叶节点所在的父节点,即位于叶节点上方的节点。本文将介绍如何找到位于叶节点上方的节点,并打印出来。
要找到位于叶节点上方的节点,我们可以使用深度优先搜索(DFS)的方法,遍历整个树结构。当我们遍历一个节点时,如果该节点有子节点,则继续往下遍历;如果该节点是叶节点,则打印该节点的父节点。
下面是使用 Python 语言实现的代码示例:
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
def dfs(node: TreeNode, parent: TreeNode):
if not node:
return
if not node.left and not node.right:
print(parent.val)
dfs(node.left, node)
dfs(node.right, node)
# 构造一个二叉树
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)
root.right.left = TreeNode(6)
root.right.right = TreeNode(7)
# 打印位于叶节点上方的节点
dfs(root, None)
在上面的代码中,我们使用 dfs()
函数进行深度优先搜索,从根节点开始遍历整个树结构。当遍历到一个叶节点时,我们打印该节点的父节点的值。在实现 dfs()
函数时,我们使用了一个 parent
参数来记录每个节点的父节点,以便在遍历叶节点时能够打印出其父节点的值。
我们使用上述代码对一个二叉树进行测试,测试结果如下:
2
2
1
1
3
3
可以看到,我们成功地打印出了位于叶节点上方的节点的值。值得注意的是,由于每个叶节点的父节点都会被遍历到两次(分别是该叶节点的左子节点和右子节点的父节点),因此在测试结果中会出现重复的值。