数据结构与算法练习（学习笔记五）

数据结构与算法

每日一题

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

示例 1：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/average-of-levels-in-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
代码如下：

from collections import deque
class Solution:
    def averageOfLevels(self, root: TreeNode) -> List[float]:
        if not root:
            return []
        ans =[]
        queue = deque([root])
        while queue:
            now_len = len(queue)
            now_levelcount = 0
            for _ in range(now_len):
                node = queue.popleft()
                now_levelcount += node.val
                if node.right:
                    queue.append(node.right)
                if node.left:
                    queue.append(node.left)
            ans.append(now_levelcount/now_len)
        return ans