每日水题 637. 二叉树的层平均值(python实现)

完美代码

于 2020-09-12 09:10:26 发布

阅读量3.1k

点赞数 1

CC 4.0 BY-SA版权

文章标签：二叉树算法数据结构 leetcode 队列

谢绝转载-https://update.blog.youkuaiyun.com

本文链接：https://blog.youkuaiyun.com/update7/article/details/108545114

编程. 同时被 2 个专栏收录

578 篇文章 ¥299.90 ¥399.90

订阅专栏

编程..

570 篇文章 ¥299.90 ¥399.90

订阅专栏

给定一个非空二叉树，通过Python实现计算每一层节点的平均值，例如输入一个二叉树，其结构为3、9、20（9的左子节点为15，右子节点为7），则输出结果为[3, 14.5, 11]，分别代表每层的平均值。" 112152780,10559304,国产AR眼镜am glass深度解析：配置、价格与显示模组,"['AR技术', '智能眼镜', '光学模组', '高通骁龙', '视觉识别']

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

给定一个非空二叉树, 返回一个由每层节点平均值组成的数组。

示例 1：

输入：
3
/ \
9 20
/ \
15 7
输出：[3, 14.5, 11]
解释：
第 0 层的平均值是 3 , 第1层是 14.5 , 第2层是 11 。因此返回 [3, 14.5, 11] 。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def averageOfLevels(self, root: TreeNode) -> List[float]:
        # 水题，直接DFS或者BFS即可，时间复杂度O(n)
        # 这里在下就使用一个队列实现之
        from queue import Queue
        q = Queue()
        self.num = [] # 记录某一层的个数
        self.cnt = [] # 记录某一层之和
        root.h = 0
        q.put(root)
        while not q.empt

了解本专栏