637. 二叉树的层平均值

最新推荐文章于 2024-08-31 14:38:26 发布

michael - z

最新推荐文章于 2024-08-31 14:38:26 发布

阅读量66

点赞数

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

本文链接：https://blog.csdn.net/zmh908264302/article/details/115536079

版权

leetcode 专栏收录该内容

81 篇文章 0 订阅

订阅专栏

文章目录

637. 二叉树的层平均值

题目描述

在这里插入图片描述

解题思路

二叉树的层次遍历

代码实现

"""
# 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: TreeNode) -> List[float]:
        stack = [root]
        res = []
        while stack:
            size = len(stack)
            _sum = 0
            for _ in range(size):
                node = stack.pop(0)
                _sum += node.val
                if node.left:
                    stack.append(node.left)
                if node.right:
                    stack.append(node.right)
            res.append(_sum / size)
        return res