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)
复杂度分析:
时间复杂度: O(N)O(N),其中 NN 为树的大小。
空间复杂度: O(N)O(N)。
参考文献:
https://leetcode-cn.com/problems/smallest-subtree-with-all-the-deepest-nodes/solution/ju-you-suo-you-zui-shen-jie-dian-de-zui-xiao-zi-sh/