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

前端之家收集整理的这篇文章主要介绍了【python-leetcode637-树的宽度遍历】二叉树的层平均值前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

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

示例 1:

输入:
3
/ \
9 20
    / \
  15 7
输出: [3,14.5,11]
解释:
第0层的平均值是 3,第1层是 14.5,第2层是 11. 因此返回 [3,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)
                 t.right:
                    queue.append(t.right)
            res.append(float(s)/float(l))
        return res

 

猜你在找的Python相关文章