leetcode_637. 二叉树的层平均值

目录

一、题目内容

二、解题思路

三、代码


一、题目内容

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

示例 1:

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

提示:

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

二、解题思路

记录每个深度的节点个数,每加一个数都求平均。

从添加的第三个数开始,平均值=原始的和* (已经加和节点个数 / (已经加和节点个数 + 1)) + 当前节点值 / (已经加和节点个数 + 1) 

三、代码

# Definition for a binary tree node.
class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

    def __repr__(self):
        return str(self.val)

class Solution:
    def averageOfLevels(self, root: TreeNode) -> list:
        self.len_res = []
        self.res = []
        def dfs(root, depth):
            if not root:
                return 0
            if depth == len(self.res):

                self.res.append(float(root.val))
                self.len_res.append(1)
            else:
                if self.len_res[depth] == 1:
                    self.res[depth] = float((self.res[depth] + root.val) / (self.len_res[depth] + 1))
                    self.len_res[depth] += 1
                elif self.len_res[depth] > 1:
                    self.res[depth] = float(self.res[depth] * self.len_res[depth] / (self.len_res[depth] + 1) + 1. * root.val / (self.len_res[depth] + 1))
                    self.len_res[depth] += 1
            dfs(root.left, depth + 1)
            dfs(root.right, depth + 1)


        dfs(root, 0)

        return self.res




if __name__ == '__main__':
    # a = TreeNode(3)
    # a.left = TreeNode(9)
    # a.right = TreeNode(20)
    # # a.right.left = TreeNode(15)
    # # a.right.right = TreeNode(7)
    # a.left.left = TreeNode(15)
    # a.left.right = TreeNode(7)
    a = TreeNode(3)
    a.left = TreeNode(1)
    a.right = TreeNode(5)
    a.left.left = TreeNode(0)
    a.left.right = TreeNode(2)
    a.right.left = TreeNode(4)
    a.right.right = TreeNode(6)

    s = Solution()
    ans = s.averageOfLevels(a)
    print(ans)
悲恋花丶无心之人 CSDN认证博客专家 深度学习 神经网络 Pytorch
计算机视觉在读研究生,熟悉Pytorch,MXNet,TensorFlow,Keras等深度学习框架,主要涉及的领域有目标检测,语义分割,超分辨率重建,行人重识别等。
个人GitHub网址为:https://github.com/nickhuang1996
©️2020 CSDN 皮肤主题: Age of Ai 设计师:meimeiellie 返回首页
实付 9.90元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值