题目地址(865. 具有所有最深节点的最小子树)

题目地址(865. 具有所有最深节点的最小子树)

https://leetcode.cn/problems/smallest-subtree-with-all-the-deepest-nodes/

题目描述

给定一个根为 root 的二叉树，每个节点的深度是 该节点到根的最短距离 。

返回包含原始树中所有 最深节点 的 最小子树 。

如果一个节点在 整个树 的任意节点之间具有最大的深度，则该节点是 最深的 。

一个节点的 子树 是该节点加上它的所有后代的集合。

 

示例 1：

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


示例 2：

输入：root = [1]
输出：[1]
解释：根节点是树中最深的节点。

示例 3：

输入：root = [0,1,3,null,2]
输出：[2]
解释：树中最深的节点为 2 ，有效子树为节点 2、1 和 0 的子树，但节点 2 的子树最小。

 

提示：

树中节点的数量在 [1, 500] 范围内。
0 <= Node.val <= 500
每个节点的值都是 独一无二 的。

 

注意：本题与力扣 1123 重复：https://leetcode-cn.com/problems/lowest-common-ancestor-of-deepest-leaves

前置知识

公司

• 暂无

思路

关键点

代码

• 语言支持：Python3

Python3 Code:

# 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 subtreeWithAllDeepest(self, root: TreeNode) -> TreeNode:
        """
        返回具有所有最深结点的最小子树，如果只有一个最深结点，那么返回叶子结点；如果有两个最深结点，返回其父节点
        Args:
            root(TreeNode):根节点
        Returns:
            最深子树
        采用带参数的dfs，返回一个元组，带有深度和结点
        """
        def dfs(root,d):
            if not root:
                return (root,d)
            l,ld = dfs(root.left,d+1)
            r,rd = dfs(root.right,d+1)
            # 如果遇到叶子结点或者左右子树深度相同返回父节点
            if ld==rd:return (root,ld) 
            return (l,ld) if ld>rd else (r,rd)
        return dfs(root,-1)[0] # 只需要返回结点信息

复杂度分析

令 n 为数组长度。

• 时间复杂度：$O(n)$
• 空间复杂度：$O(1)$