Leetcode637. 二叉树的层平均值 Average of Levels in Binary Tree - Python 以递归/迭代算法实现

最新推荐文章于 2024-07-20 22:34:59 发布

princey2100

最新推荐文章于 2024-07-20 22:34:59 发布

阅读量654

点赞数

分类专栏： leetcode 力扣二叉树文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/rendalian/article/details/127025202

版权

leetcode 同时被 3 个专栏收录

69 篇文章 0 订阅

订阅专栏

力扣

47 篇文章 0 订阅

订阅专栏

二叉树

35 篇文章 0 订阅

订阅专栏

class Solution:
    def averageOfLevels(self, root: Optional[TreeNode]) -> List[float]:
        result = []

        def traversal(root: TreeNode(), depth):
            if len(result) == depth: result.append([])
            result[depth].append(root.val)
            if root.left: traversal(root.left, depth + 1)
            if root.right: traversal(root.right, depth + 1)
        
        traversal(root, 0)
        res = []
        import statistics
        for i in range(len(result)):
            res.append(statistics.mean(result[i]))
        return res

递归思路：

很简单，就等于层序遍历二叉树