算法:二叉树的层平均值

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

输入:
    3
   / \
  9  20
    /  \
   15   7
输出:[3, 14.5, 11]

题解
也就是遍历树结构,广度优先。
1,用一个数组,将第一层都装进去。
2,将此数组求平均值
3,用当前数组的每一个元素,获取下一层的元素,装入数组中。只切当前一层的元素
重复以上步骤

代码

# 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]:

        head = root
        level = [head]
        res = []

        while level:
            sumLevel = 0
            size = len(level)
            for i in level:
                if i:
                    sumLevel += i.val
            res.append(float(sumLevel/size))
            
            for i in level[:size]:
                if i:
                    left = i.left
                    right = i.right
                    if left:
                        level.append(left)
                    if right:
                        level.append(right)
            level = level[size:]
        return res


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值