📜  打印位于叶节点上方的节点(1)

📅  最后修改于: 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

可以看到,我们成功地打印出了位于叶节点上方的节点的值。值得注意的是,由于每个叶节点的父节点都会被遍历到两次(分别是该叶节点的左子节点和右子节点的父节点),因此在测试结果中会出现重复的值。