Python每日一练:二叉树的层平均值(LeetCode)

给定一个非空二叉树的根节点 root , 以数组的形式返回每一层节点的平均值。与实际答案相差 10-5 以内的答案可以被接受

示例 1:

输入:root = [3,9,20,null,null,15,7]
输出:[3.00000,14.50000,11.00000]
解释:第 0 层的平均值为 3,第 1 层的平均值为 14.5,第 2 层的平均值为 11 。
因此返回 [3, 14.5, 11] 。

题目分析:

        题目简单易懂就是计算每层的平均值,计算平均值就要计算该层每个值的和,该层值的个数。root是一个TreeNode对象,它不支持使用下标进行索引操作,所以要用队列来遍历

代码:

#这里是树节点的结构
# 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

#从collections模块中导入deque,它是一个双端队列,支持在两端高效地进行插入和删除操作
from collections import deque
class Solution:
    def averageOfLevels(self, root: Optional[TreeNode]) -> List[float]:
        average = []    #创建一个空列表,用于存储每一层的平均值
        queue = deque([root])    #创建一个双端队列,并将根节点root放入其中。使用队列来进行层次遍历
        while queue:
            #用于计算当前层节点值的总和,并获取当前队列的长度作为level_size
            level_sum = 0
            level_size = len(queue)
            for _ in range(level_size):
                node = queue.popleft()    #弹出队列的左侧,得到当前层的一个节点node
                level_sum += node.val    #将当前节点的值加到level_sum中,累积得到当前层节点值的总和
                # 如果当前节点有左子节点或右子节点,则将它们依次加入队列中,以便后续处理
                if node.left:
                    queue.append(node.left)
                if node.right:
                    queue.append(node.right)
            average.append(level_sum / level_size)
        return average 

涉及内容:

  这里涉及到数据结构的二叉树队列深度优先搜索广度优先搜索等知识点,本篇文章使用广度优先搜索来完成,因为它相较深度优先搜索更加容易理解,关于其他几个知识点我会特别再写一篇来介绍以及解释

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值