给定一个非空二叉树的根节点 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
涉及内容:
这里涉及到数据结构的二叉树、队列、深度优先搜索和广度优先搜索等知识点,本篇文章使用广度优先搜索来完成,因为它相较深度优先搜索更加容易理解,关于其他几个知识点我会特别再写一篇来介绍以及解释