【python-leetcode637-树的宽度遍历】二叉树的层平均值

时间:2022-07-23
本文章向大家介绍【python-leetcode637-树的宽度遍历】二叉树的层平均值,主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。

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

示例 1:

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

节点值的范围在32位有符号整数范围内。

# 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]:
        if  not root:
            return []
        queue=[root]
        res=[]
        while queue:
            s=0
            l=len(queue)
            for i in range(l):
                t=queue.pop(0)
                s+=t.val
                if t.left:
                    queue.append(t.left)
                if t.right:
                    queue.append(t.right)
            res.append(float(s)/float(l))
        return res