637二叉树的层平均值

来颗花椰菜

于 2022-08-24 21:37:39 发布

阅读量71

点赞数

分类专栏：二叉树文章标签： leetcode python 算法

本文链接：https://blog.csdn.net/qq_44658259/article/details/126513256

版权

二叉树专栏收录该内容

32 篇文章 0 订阅

订阅专栏

637二叉树的层平均值

给定一个非空二叉树的根节点 root , 以数组的形式返回每一层节点的平均值。与实际答案相差 10-5 以内的答案可以被接受。

输入：root = [3,9,20,null,null,15,7]
输出：[3.00000,14.50000,11.00000]
解释：第 0 层的平均值为 3,第 1 层的平均值为 14.5,第 2 层的平均值为 11 。
因此返回 [3, 14.5, 11] 。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def averageOfLevels(self, root: Optional[TreeNode]) -> List[float]:
        #非递归
        # results = []
        # from collections import deque
        # que = deque([root])
        # while que:
        #     result = []
        #     for i in range(len(que)):
        #         cur = que.popleft()
        #         result.append(cur.val)
        #         if cur.left:
        #             que.append(cur.left)
        #         if cur.right:
        #             que.append(cur.right)
        #     results.append(sum(result)/len(result))        
        # return results
        #递归
        res = []
        def helper(root, depth):
            if len(res) == depth:res.append([])
            res[depth].append(root.val)
            if root.left:helper(root.left, depth+1)
            if root.right:helper(root.right, depth+1)
        
        helper(root, 0)
        result = []
        for i in res:
            result.append(sum(i)/len(i))
        return result