LeetCode-Python-1120. 子树的最大平均值

最新推荐文章于 2024-06-30 22:29:20 发布

暴躁老哥在线刷题

最新推荐文章于 2024-06-30 22:29:20 发布

阅读量2.8k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Leetcode 文章标签：树递归中序遍历

本文链接：https://blog.youkuaiyun.com/qq_32424059/article/details/95805733

Leetcode 同时被 2 个专栏收录

821 篇文章

订阅专栏

Python

664 篇文章

订阅专栏

博客围绕找出二叉树每一棵子树平均值中的最大值展开。子树由任意节点及其后代构成，树的平均值为节点值总和除以节点数。通过递归和中序遍历找到当前子树节点值与个数，计算平均值并与左右子树结果对比，返回最大值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给你一棵二叉树的根节点 root，找出这棵树的 每一棵 子树的 平均值 中的最大值。

子树是树中的任意节点和它的所有后代构成的集合。

树的平均值是树中节点值的总和除以节点数。

示例：

输入：[5,6,1]
输出：6.00000
解释： 
以 value = 5 的节点作为子树的根节点，得到的平均值为 (5 + 6 + 1) / 3 = 4。
以 value = 6 的节点作为子树的根节点，得到的平均值为 6 / 1 = 6。
以 value = 1 的节点作为子树的根节点，得到的平均值为 1 / 1 = 1。
所以答案取最大值 6。

提示：

树中的节点数介于 1 到 5000之间。
每个节点的值介于 0 到 100000 之间。
如果结果与标准答案的误差不超过 10^-5，那么该结果将被视为正确答案。

思路：

树的问题上递归，

中序遍历找到当前子树的所有节点的值，以及当前字数所有节点的个数，

因而可以计算出当前子树的平均值，

然后把刚刚算出来的平均值跟左右子树的结果对比一下，返回最大值即可。

class Solution(object):
    def maximumAverageSubtree(self, root):
        """
        :type root: TreeNode
        :rtype: float
        """
        if not root:
            return 0
        
        def inorder(node):
            if not node:
                return []
            return inorder(node.left) + [node.val] + inorder(node.right)
        
        Inorder = inorder(root)
        return max(1.0 * sum(Inorder) / len(Inorder), self.maximumAverageSubtree(root.left), self.maximumAverageSubtree(root.right))