1123. 最深叶节点的最近公共祖先

最新推荐文章于 2023-09-06 13:14:56 发布

Sun_Sky_Sea

最新推荐文章于 2023-09-06 13:14:56 发布

阅读量151

点赞数

分类专栏： # 树文章标签： leetcode

本文链接：https://blog.csdn.net/u013243296/article/details/113946957

版权

树专栏收录该内容

38 篇文章 0 订阅

订阅专栏

1123. 最深叶节点的最近公共祖先

原始题目链接：https://leetcode-cn.com/problems/lowest-common-ancestor-of-deepest-leaves/

给你一个有根节点的二叉树，找到它最深的叶节点的最近公共祖先。
在这里插入图片描述

输入：root = [3,5,1,6,2,0,8,null,null,7,4]
输出：[2,7,4]
解释：我们返回值为 2 的节点，在图中用黄色标记。在图中用蓝色标记的是树的最深的节点。注意，节点 6、0 和 8 也是叶节点，但是它们的深度是 2 ，而节点 7 和 4 的深度是 3 。

示例 2：

输入：root = [1]
输出：[1]
解释：根节点是树中最深的节点，它是它本身的最近公共祖先。

示例 3：

输入：root = [0,1,3,null,2]
输出：[2]
解释：树中最深的叶节点是 2 ，最近公共祖先是它自己。

解题思路：

双递归：两次深度优先搜索，第一次深度优先搜索标记树中所有节点的深度，再找到最深的节点；第二次深度优先搜索找到符合条件的最深的叶节点的最近公共祖先。

代码实现：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def lcaDeepestLeaves(self, root: TreeNode) -> TreeNode:
        # 双递归方法，第一次递归标记树中所有节点的深度
        
        # 初始化深度字典：键是节点，值是对应的深度
        depth = {None: -1}
       
        # 第一次递归，深度优先遍历：标记树中所有节点的深度
        def dfs(root, parent=None):
            if root:
                depth[root] = depth[parent] + 1
                # 计算左子节点
                dfs(root.left, root)
                # 计算右子节点
                dfs(root.right, root)
        
        # 调用dfs函数
        dfs(root)

        # 统计节点的最大深度
        max_depth = max(depth.values())

        # 第二次递归，同样适用深度优先遍历，查找最终答案
        def dfs_answer(root):
            if not root or depth.get(root) == max_depth:
                return root
            left, right = dfs_answer(root.left), dfs_answer(root.right)
            return root if left and right else left or right

        return dfs_answer(root)