leetcode-865. 具有所有最深结点的最小子树

最新推荐文章于 2022-03-25 19:59:30 发布

AI_Younger_Man

最新推荐文章于 2022-03-25 19:59:30 发布

阅读量183

点赞数

分类专栏： # Leetcode

本文链接：https://blog.csdn.net/qq_38888209/article/details/108879128

版权

Leetcode 专栏收录该内容

84 篇文章 0 订阅

订阅专栏

具有所有最深结点的最小子树
给定一个根为 root 的二叉树，每个结点的深度是它到根的最短距离。

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

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

返回能满足“以该结点为根的子树中包含所有最深的结点”这一条件的具有最大深度的结点。

示例：
在这里插入图片描述

输入：[3,5,1,6,2,0,8,null,null,7,4]
输出：[2,7,4]
解释：

我们返回值为 2 的结点，在图中用黄色标记。
在图中用蓝色标记的是树的最深的结点。
输入 “[3, 5, 1, 6, 2, 0, 8, null, null, 7, 4]” 是对给定的树的序列化表述。
输出 “[2, 7, 4]” 是对根结点的值为 2 的子树的序列化表述。
输入和输出都具有 TreeNode 类型。

思路：
树的结构，首先考虑吧能不能递归。这道题目可以递归。
首先定义递归结束条件，空结点直接返回（0 ， 0）（当前结点高度，左右子树高度大的结点）

在这里插入图片描述

# 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:

        def func(root: TreeNode):
            if not root:  # 空结点
                return 0, None   # (当前结点的高度，当前节点左右子树最大高度的结点)
            if not root.left and not root.right:  # 叶子节点
                return 0, root
            
            # 当前叶子节点的最优解
            left_high, left_node = func(root.left)  
            right_high, right_node = func(root.right)

            cur_ans = left_node if left_high > right_high else right_node

            if left_high == right_high:
                return left_high+1, root
            elif left_high > right_high:
                return left_high+1, cur_ans
            else:
                return right_high+1, cur_ans

        my_high, ret = func(root)
        return ret